Add GPU version for yolo3detection

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-03-23 21:29:29 +01:00
parent 3bc648c07b
commit 785a3d8dd5
3 changed files with 11 additions and 5 deletions
+2
View File
@@ -17,6 +17,8 @@ private:
tk::dnn::Yolo* yolo[3];
tk::dnn::Yolo* getYoloLayer(int n=0);
cv::Mat bgr_h;
public:
Yolo3Detection() {};