summaryrefslogtreecommitdiff
path: root/source/application/api/use_case/object_detection/include/YoloFastestModel.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'source/application/api/use_case/object_detection/include/YoloFastestModel.hpp')
-rw-r--r--source/application/api/use_case/object_detection/include/YoloFastestModel.hpp13
1 files changed, 8 insertions, 5 deletions
diff --git a/source/application/api/use_case/object_detection/include/YoloFastestModel.hpp b/source/application/api/use_case/object_detection/include/YoloFastestModel.hpp
index 4c64433..16d0715 100644
--- a/source/application/api/use_case/object_detection/include/YoloFastestModel.hpp
+++ b/source/application/api/use_case/object_detection/include/YoloFastestModel.hpp
@@ -19,13 +19,16 @@
#include "Model.hpp"
-extern const int originalImageSize;
-extern const int channelsImageDisplayed;
-extern const float anchor1[];
-extern const float anchor2[];
-
namespace arm {
namespace app {
+ namespace object_detection {
+ extern const int originalImageSize;
+ extern const int channelsImageDisplayed;
+ /* NOTE: anchors are different for any given input model size, estimated during training
+ * phase */
+ extern const float anchor1[];
+ extern const float anchor2[];
+ } /* namespace object_detection */
class YoloFastestModel : public Model {