Modified resize and boxes coordinates to float, to achieve same darknet accuracy

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-08-05 17:25:33 +02:00
parent a5d2d4792a
commit 8ff7cad2e1
3 changed files with 112 additions and 33 deletions
+2 -1
View File
@@ -13,6 +13,7 @@ private:
int num = 0;
int nMasks = 0;
int nDets = 0;
bool letterbox = false;
tk::dnn::Yolo::detection *dets = nullptr;
tk::dnn::Yolo* yolo[3];
@@ -21,7 +22,7 @@ private:
cv::Mat bgr_h;
public:
Yolo3Detection() {};
Yolo3Detection(const bool letter_box=false) :letterbox(letter_box){}
~Yolo3Detection() {};
bool init(const std::string& tensor_path, const int n_classes=80, const int n_batches=1);