Skip to content

Commit

Permalink
updates
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Apr 29, 2019
1 parent 94f954e commit 587522e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,7 @@ def __init__(self, path, img_size=416, batch_size=16, augment=False):
for x in self.img_files]

# Rectangular Training https://github.com/ultralytics/yolov3/issues/232
self.train_rectangular = True
self.train_rectangular = False
if self.train_rectangular:
bi = np.floor(np.arange(n) / batch_size).astype(np.int) # batch index
nb = bi[-1] + 1 # number of batches
Expand Down

0 comments on commit 587522e

Please sign in to comment.