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

Repair convert_coco() box-segment discarding (#6689)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ileal16 2023-11-30 13:49:24 +01:00 • committed by GitHub
parent 2e71f7f50e
commit 16639b60eb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/__init__.py
View file

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = '8.0.220' __version__ = '8.0.221'
from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models import RTDETR, SAM, YOLO
from ultralytics.models.fastsam import FastSAM from ultralytics.models.fastsam import FastSAM

1
ultralytics/data/converter.py
View file

@ -129,7 +129,6 @@ def convert_coco(labels_dir='../coco/annotations/',
s = [j for i in ann['segmentation'] for j in i] # all segments concatenated s = [j for i in ann['segmentation'] for j in i] # all segments concatenated
s = (np.array(s).reshape(-1, 2) / np.array([w, h])).reshape(-1).tolist() s = (np.array(s).reshape(-1, 2) / np.array([w, h])).reshape(-1).tolist()
s = [cls] + s s = [cls] + s
if s not in segments:
segments.append(s) segments.append(s)
if use_keypoints and ann.get('keypoints') is not None: if use_keypoints and ann.get('keypoints') is not None:
keypoints.append(box + (np.array(ann['keypoints']).reshape(-1, 3) / keypoints.append(box + (np.array(ann['keypoints']).reshape(-1, 3) /