diff options
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.hpp | 13 |
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 { |