--rect test bug fix

This commit is contained in:
Glenn Jocher committed 2020-07-09 20:39:11 -07:00
1 parent d40e4781f1
commit 6649770001
2 files changed
+2 -1

No files matched your search

+1
View File
@@ -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]