xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix __len__ and single_cls (#249)

Browse Source
This commit is contained in:
Laughing 2023-01-11 18:25:10 +08:00 committed by GitHub
parent 88c9418087
commit 53fa0f4550
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/yolo/data/base.py
View File

@ -110,7 +110,7 @@ class BaseDataset(Dataset):
if segments: if segments:
self.labels[i]["segments"] = segments[j] self.labels[i]["segments"] = segments[j]
if self.single_cls: if self.single_cls:
self.labels[i]["cls"] = 0 self.labels[i]["cls"][:, 0] = 0
def load_image(self, i): def load_image(self, i):
# Loads 1 image from dataset index 'i', returns (im, resized hw) # Loads 1 image from dataset index 'i', returns (im, resized hw)
@ -191,7 +191,7 @@ class BaseDataset(Dataset):
return label return label
def __len__(self): def __len__(self):
return len(self.im_files) return len(self.labels)
def update_labels_info(self, label): def update_labels_info(self, label):
"""custom your label format here""" """custom your label format here"""