dlr/predict typo

This commit is contained in:
Starbeamrainbowlabs 2023-01-12 18:03:06 +00:00
parent 0140e93bbd
commit 864dfa802d
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -247,7 +247,7 @@ def get_overlay(image, coloured_mask):
def plot_samples_matplotlib(filepath, display_list, figsize=(5, 3)):
for i in range(len(display_list)):
plt.subplot(1, len(display_list), i)
plt.subplot(1, len(display_list), i+1)
if display_list[i].shape[-1] == 3:
plt.imshow(tf.keras.preprocessing.image.array_to_img(display_list[i]))
else: