Actually call yolov5 inference
Signed-off-by: Filip Kujawa <filip.j.kujawa@gmail.com>
Change-Id: I693aa253d09e88f6000ee9ea5a8c68862ac47629
diff --git a/y2023/vision/game_pieces.h b/y2023/vision/game_pieces.h
index a41d52a..a99a99b 100644
--- a/y2023/vision/game_pieces.h
+++ b/y2023/vision/game_pieces.h
@@ -5,6 +5,8 @@
#include "frc971/vision/vision_generated.h"
#include "y2023/vision/game_pieces_generated.h"
+#include "y2023/vision/yolov5.h"
+
namespace y2023 {
namespace vision {
@@ -20,6 +22,7 @@
private:
aos::Sender<GamePieces> game_pieces_sender_;
+ std::unique_ptr<YOLOV5> model;
};
} // namespace vision
} // namespace y2023