[decoder] Update yolov5 decoder parameters
authorYongjoo Ahn <yongjoo1.ahn@samsung.com>
Fri, 16 Dec 2022 06:27:27 +0000 (15:27 +0900)
committerMyungJoo Ham <myungjoo.ham@samsung.com>
Fri, 23 Dec 2022 09:23:21 +0000 (18:23 +0900)
- Set the conf and iou threshold values to follow the upstream
  https://github.com/ultralytics/yolov5.

Signed-off-by: Yongjoo Ahn <yongjoo1.ahn@samsung.com>
ext/nnstreamer/tensor_decoder/tensordec-boundingbox.c

index 9be94a8..39c3478 100644 (file)
@@ -123,8 +123,8 @@ extern uint8_t rasters[][13];
 #define OV_PERSON_DETECTION_SIZE_DETECTION_DESC (7)
 #define OV_PERSON_DETECTION_CONF_THRESHOLD      (0.8)
 #define YOLOV5_DETECTION_NUM_INFO               (5)
-#define YOLOV5_DETECTION_CONF_THRESHOLD         (0.3)
-#define YOLOV5_DETECTION_IOU_THRESHOLD          (0.6)
+#define YOLOV5_DETECTION_CONF_THRESHOLD         (0.25)
+#define YOLOV5_DETECTION_IOU_THRESHOLD          (0.45)
 #define PIXEL_VALUE                             (0xFF0000FF)    /* RED 100% in RGBA */
 #define MP_PALM_DETECTION_INFO_SIZE             (18)
 #define MP_PALM_DETECTION_MAX_TENSORS           (2U)