Zhen Ye commited on
Commit
4e93b33
·
1 Parent(s): af29397

fix: Resolve IndentationError in DroneYoloDetector due to displaced import

Browse files
Files changed (1) hide show
  1. models/detectors/drone_yolo.py +1 -1
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."""