sem/SDBA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[EDIT] stuff

Browse Source
This commit is contained in:
Sem van der Hoeven
2021-06-04 11:39:23 +02:00
parent fe94b0f83d
commit 921609de5d
3 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/computervision/OpenPoseVideo.cpp
View File

@@ -42,9 +42,11 @@ namespace computervision
void OpenPoseVideo::setup() {
net = readNetFromCaffe(protoFile, weightsFile);
net.setPreferableBackend(DNN_TARGET_CPU);
}
void OpenPoseVideo::movementSkeleton(Mat inputImage, std::function<void(std::vector<Point>,cv::Mat poinst_on_image)> f) {
void OpenPoseVideo::movementSkeleton(Mat& inputImage, std::function<void(std::vector<Point>&, cv::Mat& poinst_on_image)> f) {
std::cout << "movement skeleton start" << std::endl;
int inWidth = 368;
@@ -67,6 +69,7 @@ namespace computervision
std::cout << "done setting input to net" << std::endl;
Mat output = net.forward();
std::cout << "time took to set input and forward: " << t << std::endl;
int H = output.size[2];
int W = output.size[3];

2
src/computervision/OpenPoseVideo.h
View File

@@ -13,7 +13,7 @@ namespace computervision
private:
public:
void movementSkeleton(Mat inputImage,std::function<void(std::vector<Point>, cv::Mat poinst_on_image)> f);
void movementSkeleton(Mat& inputImage, std::function<void(std::vector<Point>&, cv::Mat& poinst_on_image)> f);
void setup();
};
}

7
src/main.cpp
View File

@@ -87,10 +87,11 @@ int main(void)
// set up object detection
//objDetect.setup();
cv::VideoCapture cam = objDetect.getCap();
//cv::VideoCapture cam = objDetect.getCap();
cv::Mat img;
cam.read(img);
imshow("camera in main loop", img);
cv::VideoCapture cap = objDetect.getCap();
//cam.read(img);
//imshow("camera in main loop", img);
computervision::AsyncArmDetection as;