Zhen Ye commited on
Commit
0719ba5
·
1 Parent(s): 301e154

chore(yolo11): update yolo naming comments

Browse files
Files changed (2) hide show
  1. utils/mission_parser.py +1 -1
  2. utils/roofline.py +2 -2
utils/mission_parser.py CHANGED
@@ -409,7 +409,7 @@ def build_broad_queries(
409
  # LLM_EXTRACTED path: detect broadly
410
  if _is_coco_only(detector_key):
411
  # COCO detectors ignore queries anyway (DETR detects all 80;
412
- # YOLOv8 falls back to all if no matches). Send everything.
413
  return list(COCO_CLASSES)
414
 
415
  # Open-vocab detector (e.g. Grounding DINO):
 
409
  # LLM_EXTRACTED path: detect broadly
410
  if _is_coco_only(detector_key):
411
  # COCO detectors ignore queries anyway (DETR detects all 80;
412
+ # YOLO11 falls back to all if no matches). Send everything.
413
  return list(COCO_CLASSES)
414
 
415
  # Open-vocab detector (e.g. Grounding DINO):
utils/roofline.py CHANGED
@@ -15,10 +15,10 @@ logger = logging.getLogger(__name__)
15
  # These are rough estimates; actual FLOPs depend on input resolution and model variant.
16
  _MODEL_FLOPS: Dict[str, float] = {
17
  # Detection models (GFLOPs per frame)
18
- "yolo11": 78.9, # YOLOv8m ~79 GFLOPs at 640px
19
  "detr_resnet50": 86.0, # DETR-R50 ~86 GFLOPs at 800px
20
  "grounding_dino": 172.0, # Grounding DINO-B ~172 GFLOPs
21
- "drone_yolo": 78.9, # Same arch as YOLOv8m
22
 
23
  # Segmentation models (GFLOPs per keyframe)
24
  "GSAM2-S": 48.0, # SAM2 small encoder
 
15
  # These are rough estimates; actual FLOPs depend on input resolution and model variant.
16
  _MODEL_FLOPS: Dict[str, float] = {
17
  # Detection models (GFLOPs per frame)
18
+ "yolo11": 78.9, # YOLO11m ~79 GFLOPs at 640px
19
  "detr_resnet50": 86.0, # DETR-R50 ~86 GFLOPs at 800px
20
  "grounding_dino": 172.0, # Grounding DINO-B ~172 GFLOPs
21
+ "drone_yolo": 78.9, # Same arch as YOLO11m-class model
22
 
23
  # Segmentation models (GFLOPs per keyframe)
24
  "GSAM2-S": 48.0, # SAM2 small encoder