Spaces:
Sleeping
Sleeping
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]
|