Skip to content

Commit bbd2c43

Browse files
committed
🔧 [Update] default confidence score & error handle
1 parent dc1a0cf commit bbd2c43

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

‎yolo/config/task/inference.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,5 +7,5 @@ data:
77
pin_memory: True
88
data_augment: {}
99
nms:
10-
min_confidence: 0.1
10+
min_confidence: 0.5
1111
min_iou: 0.5

‎yolo/tools/solver.py‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,9 +118,10 @@ def solve(self, dataloader: StreamDataLoader):
118118
images[0], nms_out[0], scaled_bbox=False, save_path=self.save_path, save_name=f"frame{idx:03d}.png"
119119
)
120120
except (KeyboardInterrupt, Exception) as e:
121-
logger.error(str(e))
122121
dataloader.stop_event.set()
123122
dataloader.stop()
124-
if not isinstance(e, KeyboardInterrupt):
123+
if isinstance(e, KeyboardInterrupt):
124+
logger.error("User Keyboard Interrupt")
125+
else:
125126
raise e
126127
dataloader.stop()

0 commit comments

Comments
 (0)