update the arg classes in prediction#

ai4os-hub · Jul 19, 2024 · 05bc7b4 · 05bc7b4
1 parent d6c611b
commit 05bc7b4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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,
  )