ultralytics 8.2.2 replace COCO128 with COCO8 (#10167)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
626309d221
commit
1110258d37
43 changed files with 154 additions and 156 deletions
|
|
@ -37,7 +37,7 @@ class YOLOv8Seg:
|
|||
self.model_height, self.model_width = [x.shape for x in self.session.get_inputs()][0][-2:]
|
||||
|
||||
# Load COCO class names
|
||||
self.classes = yaml_load(check_yaml("coco128.yaml"))["names"]
|
||||
self.classes = yaml_load(check_yaml("coco8.yaml"))["names"]
|
||||
|
||||
# Create color palette
|
||||
self.color_palette = Colors()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue