INNER CODE UNIT · Python
TaskSpec
ultralytics/yolo-flutter-app · scripts/export-tflite-models.py:39
class TaskSpec:
"""Export settings for one YOLO task family."""
suffix: str
imgsz: int
TASKS: dict[str, TaskSpec] = {
"detect": TaskSpec("", 640),
"segment": TaskSpec("-seg", 640),
"semantic": TaskSpec("-sem", 640),
"depth": TaskSpec("-depth", 640),
"classify": TaskSpec("-cls", 224),
"pose": TaskSpec("-pose", 640),
"obb": TaskSpec("-obb", 640),
}
_TASK_NAMES_CACHE: dict[str, dict[int, str]] = {}