Explorar o código

Updated model to 240613.Threshold change to 0.6

Qiuyang hai 5 meses
pai
achega
36c475c89a

+ 2 - 2
source/CNStream/configs/yolov5_object_detection_mlu220.json

@@ -4,7 +4,7 @@
     "parallelism" : 2,
     "max_input_queue_size" : 20,
     "custom_params" : {
-      "model_path" : "../data/models/yolov5sm_best8_flock_s11_38_600epochs_int16_20240228_4b_4c.cambricon",
+      "model_path" : "../data/models/yolov5sm_best8_flock_s13_38_900epochs_int16_20240613_4b_4c.cambricon",
       "func_name" : "subnet0",
       // Uncomment the following 2 lines to use cncv
       // "preproc_name" : "CNCV",
@@ -14,7 +14,7 @@
       "keep_aspect_ratio" : "true",
       "model_input_pixel_format" : "RGB24",
       "batching_timeout" : 50,
-      "threshold" : 0.5,
+      "threshold" : 0.6,
       "infer_interval": 6,
       "engine_num" : 2,
       "device_id" : 0

+ 2 - 1
source/CNStream/data/models/label_map_gsd.txt

@@ -1,4 +1,5 @@
 bird
 plane
 insects
-flock
+flock
+drone

BIN=BIN
source/CNStream/data/models/yolov5sm_best8_flock_s13_38_900epochs_int16_20240613_4b_4c.cambricon