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 dataset (#167)

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

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

@ -50,7 +50,7 @@ class BaseDataset(Dataset):
if self.single_cls:
self.update_labels(include_class=[])
self.ni = len(self.im_files)
self.ni = len(self.labels)
# rect stuff
self.rect = rect