--rect test bug fix
This commit is contained in:
1 parent
d40e4781f1
commit
6649770001
2 files changed
+2
-1
No files matched your search
@@ -343,6 +343,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||
irect = ar.argsort()
|
||||
self.img_files = [self.img_files[i] for i in irect]
|
||||
self.label_files = [self.label_files[i] for i in irect]
|
||||
self.labels = [self.labels[i] for i in irect]
|
||||
self.shapes = s[irect] # wh
|
||||
ar = ar[irect]
|
||||
|
||||
|
||||
Reference in new issue
Block a user