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

updates

This commit is contained in:
glenn-jocher 2019-07-03 16:17:46 +02:00
parent ab141fcc1f
commit 1e62ee2152
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -128,7 +128,7 @@ def train(
# plt.savefig('LR.png', dpi=300) # plt.savefig('LR.png', dpi=300)
# Dataset # Dataset
rectangular_training = False rectangular_training = True
dataset = LoadImagesAndLabels(train_path, dataset = LoadImagesAndLabels(train_path,
img_size, img_size,
batch_size, batch_size,
@ -196,7 +196,7 @@ def train(
imgs = imgs.to(device) imgs = imgs.to(device)
targets = targets.to(device) targets = targets.to(device)
# Multi-Scale training # Multi-Scale training TODO: short-side to 32-multiple https://github.com/ultralytics/yolov3/issues/358
if multi_scale: if multi_scale:
if (i + nb * epoch) / accumulate % 10 == 0: #  adjust (67% - 150%) every 10 batches if (i + nb * epoch) / accumulate % 10 == 0: #  adjust (67% - 150%) every 10 batches
img_size = random.choice(range(img_size_min, img_size_max + 1)) * 32 img_size = random.choice(range(img_size_min, img_size_max + 1)) * 32