Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-24 21:37:32 +02:00
parent fa0acebe2a
commit 3e71b8d48b
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
utils/datasets.py
View File

@ -144,14 +144,15 @@ class LoadImagesAndLabels(Dataset): # for training/testing
for x in self.img_files] for x in self.img_files]
# Rectangular Training https://github.com/ultralytics/yolov3/issues/232 # Rectangular Training https://github.com/ultralytics/yolov3/issues/232
self.train_rectangular = False self.train_rectangular = True
if self.train_rectangular: if self.train_rectangular:
bi = np.floor(np.arange(n) / batch_size).astype(np.int) # batch index bi = np.floor(np.arange(n) / batch_size).astype(np.int) # batch index
nb = bi[-1] # number of batches nb = bi[-1] + 1 # number of batches
from PIL import Image from PIL import Image
# Read image aspect ratios # Read image aspect ratios
s = np.array([Image.open(f).size for f in tqdm(self.img_files, desc='Reading image shapes')]) iter = tqdm(self.img_files, desc='Reading image shapes') if n > 100 else self.img_files
s = np.array([Image.open(f).size for f in iter])
ar = s[:, 1] / s[:, 0] # aspect ratio ar = s[:, 1] / s[:, 0] # aspect ratio
# Sort by aspect ratio # Sort by aspect ratio