🔧 [Update] default confidence score & error handle
Browse files- yolo/config/task/inference.yaml +1 -1
- yolo/tools/solver.py +3 -2
yolo/config/task/inference.yaml
CHANGED
@@ -7,5 +7,5 @@ data:
|
|
7 |
pin_memory: True
|
8 |
data_augment: {}
|
9 |
nms:
|
10 |
-
min_confidence: 0.
|
11 |
min_iou: 0.5
|
|
|
7 |
pin_memory: True
|
8 |
data_augment: {}
|
9 |
nms:
|
10 |
+
min_confidence: 0.5
|
11 |
min_iou: 0.5
|
yolo/tools/solver.py
CHANGED
@@ -118,9 +118,10 @@ class ModelTester:
|
|
118 |
images[0], nms_out[0], scaled_bbox=False, save_path=self.save_path, save_name=f"frame{idx:03d}.png"
|
119 |
)
|
120 |
except (KeyboardInterrupt, Exception) as e:
|
121 |
-
logger.error(str(e))
|
122 |
dataloader.stop_event.set()
|
123 |
dataloader.stop()
|
124 |
-
if
|
|
|
|
|
125 |
raise e
|
126 |
dataloader.stop()
|
|
|
118 |
images[0], nms_out[0], scaled_bbox=False, save_path=self.save_path, save_name=f"frame{idx:03d}.png"
|
119 |
)
|
120 |
except (KeyboardInterrupt, Exception) as e:
|
|
|
121 |
dataloader.stop_event.set()
|
122 |
dataloader.stop()
|
123 |
+
if isinstance(e, KeyboardInterrupt):
|
124 |
+
logger.error("User Keyboard Interrupt")
|
125 |
+
else:
|
126 |
raise e
|
127 |
dataloader.stop()
|