This commit is contained in:
glenn-jocher 2019-07-03 16:18:08 +02:00
parent 1e62ee2152
commit 109991198c
1 changed files with 1 additions and 1 deletions

View File

@ -128,7 +128,7 @@ def train(
# plt.savefig('LR.png', dpi=300) # plt.savefig('LR.png', dpi=300)
# Dataset # Dataset
rectangular_training = True rectangular_training = False
dataset = LoadImagesAndLabels(train_path, dataset = LoadImagesAndLabels(train_path,
img_size, img_size,
batch_size, batch_size,