diff --git a/api/__init__.py b/api/__init__.py index a14b4ab..91d5177 100644 --- a/api/__init__.py +++ b/api/__init__.py @@ -315,6 +315,7 @@ def main(): args = cmd_parser.parse_args() main() + """ python3 api/__init__.py train --model yolov8n.yaml\ diff --git a/api/schemas.py b/api/schemas.py index 9b5ec99..746525e 100644 --- a/api/schemas.py +++ b/api/schemas.py @@ -135,10 +135,13 @@ class Meta: }, load_default=False, ) - classes = fields.Field( + classes = fields.List( + fields.Int(), metadata={ "description": "Filter results by class, i.e. class=0, " - "or class=[0,2,3]" + "or class=[0,2,3]. Only detections belonging to the " + "specified classes will be returned. Useful for focusing" + " on relevant objects in multi-class detection tasks." }, load_default=None, )