HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.2.2 replace COCO128 with COCO8 (#10167)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-04-18 20:47:21 -07:00 • committed by GitHub
parent 626309d221
commit 1110258d37
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
43 changed files with 154 additions and 156 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/YOLOv8-ONNXRuntime/main.py
View file

@ -30,7 +30,7 @@ class YOLOv8:
self.iou_thres = iou_thres
# Load the class names from the COCO dataset
self.classes = yaml_load(check_yaml("coco128.yaml"))["names"]
self.classes = yaml_load(check_yaml("coco8.yaml"))["names"]
# Generate a color palette for the classes
self.color_palette = np.random.uniform(0, 255, size=(len(self.classes), 3))