Spaces:
Sleeping
Sleeping
Zhen Ye
commited on
Commit
·
4e93b33
1
Parent(s):
af29397
fix: Resolve IndentationError in DroneYoloDetector due to displaced import
Browse files
models/detectors/drone_yolo.py
CHANGED
|
@@ -7,6 +7,7 @@ import torch
|
|
| 7 |
from ultralytics import YOLO
|
| 8 |
|
| 9 |
from models.detectors.base import DetectionResult, ObjectDetector
|
|
|
|
| 10 |
|
| 11 |
|
| 12 |
class DroneYoloDetector(ObjectDetector):
|
|
@@ -62,7 +63,6 @@ class DroneYoloDetector(ObjectDetector):
|
|
| 62 |
label_names=label_names,
|
| 63 |
)
|
| 64 |
|
| 65 |
-
from utils.tiling import get_slice_bboxes, slice_image, shift_bboxes, batched_nms
|
| 66 |
|
| 67 |
def _predict_tiled(self, frame: np.ndarray, queries: Sequence[str]) -> DetectionResult:
|
| 68 |
"""Run tiled inference for high-resolution frames."""
|
|
|
|
| 7 |
from ultralytics import YOLO
|
| 8 |
|
| 9 |
from models.detectors.base import DetectionResult, ObjectDetector
|
| 10 |
+
from utils.tiling import get_slice_bboxes, slice_image, shift_bboxes, batched_nms
|
| 11 |
|
| 12 |
|
| 13 |
class DroneYoloDetector(ObjectDetector):
|
|
|
|
| 63 |
label_names=label_names,
|
| 64 |
)
|
| 65 |
|
|
|
|
| 66 |
|
| 67 |
def _predict_tiled(self, frame: np.ndarray, queries: Sequence[str]) -> DetectionResult:
|
| 68 |
"""Run tiled inference for high-resolution frames."""
|