Zhen Ye commited on
Commit
284ce20
·
1 Parent(s): a5f8d15

optimize: Force high-res (1280px) inference for YOLO models to improve small object detection

Browse files
models/detectors/drone_yolo.py CHANGED
@@ -68,6 +68,7 @@ class DroneYoloDetector(ObjectDetector):
68
  source=frame,
69
  device=device_arg,
70
  conf=self.score_threshold,
 
71
  verbose=False,
72
  )
73
  return self._parse_single_result(results[0], queries)
@@ -77,6 +78,7 @@ class DroneYoloDetector(ObjectDetector):
77
  source=frames,
78
  device=self.device,
79
  conf=self.score_threshold,
 
80
  verbose=False,
81
  )
82
  return [self._parse_single_result(r, queries) for r in results]
 
68
  source=frame,
69
  device=device_arg,
70
  conf=self.score_threshold,
71
+ imgsz=1280,
72
  verbose=False,
73
  )
74
  return self._parse_single_result(results[0], queries)
 
78
  source=frames,
79
  device=self.device,
80
  conf=self.score_threshold,
81
+ imgsz=1280,
82
  verbose=False,
83
  )
84
  return [self._parse_single_result(r, queries) for r in results]
models/detectors/yolov8.py CHANGED
@@ -69,6 +69,7 @@ class HuggingFaceYoloV8Detector(ObjectDetector):
69
  source=frame,
70
  device=self.device,
71
  conf=self.score_threshold,
 
72
  verbose=False,
73
  )
74
  return self._parse_single_result(results[0], queries)
@@ -78,6 +79,7 @@ class HuggingFaceYoloV8Detector(ObjectDetector):
78
  source=frames,
79
  device=self.device,
80
  conf=self.score_threshold,
 
81
  verbose=False,
82
  )
83
  return [self._parse_single_result(r, queries) for r in results]
 
69
  source=frame,
70
  device=self.device,
71
  conf=self.score_threshold,
72
+ imgsz=1280,
73
  verbose=False,
74
  )
75
  return self._parse_single_result(results[0], queries)
 
79
  source=frames,
80
  device=self.device,
81
  conf=self.score_threshold,
82
+ imgsz=1280,
83
  verbose=False,
84
  )
85
  return [self._parse_single_result(r, queries) for r in results]