henry000 commited on
Commit
bbd2c43
·
1 Parent(s): dc1a0cf

🔧 [Update] default confidence score & error handle

Browse files
yolo/config/task/inference.yaml CHANGED
@@ -7,5 +7,5 @@ data:
7
  pin_memory: True
8
  data_augment: {}
9
  nms:
10
- min_confidence: 0.1
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 not isinstance(e, KeyboardInterrupt):
 
 
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()