xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests 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
1 changed files with 1 additions and 1 deletions
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: if self.single_cls:
self.update_labels(include_class=[]) self.update_labels(include_class=[])
self.ni = len(self.im_files) self.ni = len(self.labels)
# rect stuff # rect stuff
self.rect = rect self.rect = rect