Spaces:
Paused
Paused
Zhen Ye commited on
Commit ·
0719ba5
1
Parent(s): 301e154
chore(yolo11): update yolo naming comments
Browse files- utils/mission_parser.py +1 -1
- 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 |
-
#
|
| 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, #
|
| 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
|
| 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
|