Actually call yolov5 inference
Signed-off-by: Filip Kujawa <filip.j.kujawa@gmail.com>
Change-Id: I693aa253d09e88f6000ee9ea5a8c68862ac47629
diff --git a/y2023/vision/yolov5.h b/y2023/vision/yolov5.h
index ad04350..7e2a521 100644
--- a/y2023/vision/yolov5.h
+++ b/y2023/vision/yolov5.h
@@ -7,7 +7,6 @@
#include <fstream>
#include <iostream>
#include <opencv2/core.hpp>
-#include <opencv2/dnn.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc.hpp>
@@ -24,14 +23,14 @@
class YOLOV5 {
public:
- virtual ~YOLOV5();
+ virtual ~YOLOV5() {}
// Takes a model path as string and loads a pre-trained
// YOLOv5 model from the specified path.
- virtual void LoadModel(const std::string path);
+ virtual void LoadModel(const std::string path) = 0;
// Takes an image and returns a Detection.
- virtual std::vector<Detection> ProcessImage(cv::Mat image);
+ virtual std::vector<Detection> ProcessImage(cv::Mat image) = 0;
};
std::unique_ptr<YOLOV5> MakeYOLOV5();