This commit is contained in:
Glenn Jocher
2019-06-12 11:50:24 +02:00
parent 9c328b1b0e
commit 5edb0ec40d
2 changed files with 6 additions and 7 deletions
+1
View File
@@ -153,6 +153,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
replace('.bmp', '.txt').
replace('.png', '.txt') for x in self.img_files]
multi_scale = False
if multi_scale:
s = img_size / 32
self.multi_scale = ((np.linspace(0.5, 1.5, nb) * s).round().astype(np.int) * 32)