Names

app.infrastructure.detection.model.names

View source code

Detection model registry and confidence thresholds for Torch and YOLO variants.

See Also

• get_available_detection_model_names_as_str — exposes registered model names
• load_detection_model — resolves registry entries to implementations

Layer: Infrastructure. Context: Implements concrete processing and I/O behind application-facing contracts.

TORCH_MODELS: dict[str, tuple[str, str]] = {'Torch FCOS_ResNet50_FPN': ('fcos_resnet50_fpn', 'FCOS_ResNet50_FPN_Weights'), 'Torch FasterRCNN_MobileNet_V3_Large_320_FPN': ('fasterrcnn_mobilenet_v3_large_320_fpn', 'FasterRCNN_MobileNet_V3_Large_320_FPN_Weights'), 'Torch FasterRCNN_MobileNet_V3_Large_FPN': ('fasterrcnn_mobilenet_v3_large_fpn', 'FasterRCNN_MobileNet_V3_Large_FPN_Weights'), 'Torch FasterRCNN_ResNet50_FPN_V2': ('fasterrcnn_resnet50_fpn_v2', 'FasterRCNN_ResNet50_FPN_V2_Weights'), 'Torch FasterRCNN_ResNet50_FPN': ('fasterrcnn_resnet50_fpn', 'FasterRCNN_ResNet50_FPN_Weights'), 'Torch RetinaNet_ResNet50_FPN_V2': ('retinanet_resnet50_fpn_v2', 'RetinaNet_ResNet50_FPN_V2_Weights'), 'Torch RetinaNet_ResNet50_FPN': ('retinanet_resnet50_fpn', 'RetinaNet_ResNet50_FPN_Weights'), 'Torch SSD300_VGG16': ('ssd300_vgg16', 'SSD300_VGG16_Weights'), 'Torch SSDLite320_MobileNet_V3_Large': ('ssdlite320_mobilenet_v3_large', 'SSDLite320_MobileNet_V3_Large_Weights')} module-attribute

ULTRALYTICS_YOLO_MODELS: dict[str, str] = {'YOLOv3-tiny': 'yolov3-tinyu.pt', 'YOLOv3': 'yolov3u.pt', 'YOLOv3-spp': 'yolov3-sppu.pt', 'YOLOv5n': 'yolov5nu.pt', 'YOLOv5s': 'yolov5su.pt', 'YOLOv5m': 'yolov5mu.pt', 'YOLOv5l': 'yolov5lu.pt', 'YOLOv5x': 'yolov5xu.pt', 'YOLOv8n': 'yolov8n.pt', 'YOLOv8s': 'yolov8s.pt', 'YOLOv8m': 'yolov8m.pt', 'YOLOv8l': 'yolov8l.pt', 'YOLOv8x': 'yolov8x.pt', 'YOLOv9t': 'yolov9t.pt', 'YOLOv9s': 'yolov9s.pt', 'YOLOv9m': 'yolov9m.pt', 'YOLOv9c': 'yolov9c.pt', 'YOLOv9e': 'yolov9e.pt', 'YOLOv10n': 'yolov10n.pt', 'YOLOv10s': 'yolov10s.pt', 'YOLOv10m': 'yolov10m.pt', 'YOLOv10b': 'yolov10b.pt', 'YOLOv10l': 'yolov10l.pt', 'YOLOv10x': 'yolov10x.pt', 'YOLO11n': 'yolo11n.pt', 'YOLO11s': 'yolo11s.pt', 'YOLO11m': 'yolo11m.pt', 'YOLO11l': 'yolo11l.pt', 'YOLO11x': 'yolo11x.pt', 'YOLO12n': 'yolo12n.pt', 'YOLO12s': 'yolo12s.pt', 'YOLO12m': 'yolo12m.pt', 'YOLO12l': 'yolo12l.pt', 'YOLO12x': 'yolo12x.pt', 'YOLOv26n': 'yolo26n.pt', 'YOLOv26s': 'yolo26s.pt', 'YOLOv26m': 'yolo26m.pt', 'YOLOv26l': 'yolo26l.pt', 'YOLOv26x': 'yolo26x.pt'} module-attribute

MTCNN_MODELS = ('MTCNN-Tensorflow', 'MTCNN-Pytorch') module-attribute

DEFAULT_TORCH_CONFIDENCE_THRESHOLD = 0.5 module-attribute

DEFAULT_YOLO_CONFIDENCE_THRESHOLD = 0.25 module-attribute

DEFAULT_MTCNN_TF_CONFIDENCE_THRESHOLD = 0.9 module-attribute

DEFAULT_MTCNN_TORCH_CONFIDENCE_THRESHOLD = 0.9 module-attribute

TORCH_MODEL_NAMES = tuple(TORCH_MODELS) module-attribute

ULTRALYTICS_MODEL_NAMES = tuple(ULTRALYTICS_YOLO_MODELS) module-attribute

DETECTION_MODEL_NAMES = (*TORCH_MODEL_NAMES, *ULTRALYTICS_MODEL_NAMES, *MTCNN_MODELS) module-attribute