Fixed path for train images

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-10 11:59:34 +02:00
parent db53c35929
commit b926d11b84
1 changed files with 1 additions and 1 deletions

View File

@ -46,7 +46,7 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray, output_path: s
for i, train_image in enumerate(images):
instances = labels[i]
image = Image.fromarray(train_image)
image.save(f"{output_path}"
image.save(f"{output_path}/"
f"train_image{str(i).zfill(nr_digits)}.png")
figure = pyplot.figure(figsize=(20, 12))