This commit is contained in:
Starbeamrainbowlabs 2023-01-12 18:21:20 +00:00
parent c0c6e81c01
commit 20f7d34fd1
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -265,7 +265,7 @@ def plot_predictions(filepath, input_items, colormap, model):
print("DEBUG:plot_predictions INFER", prediction_mask)
plot_samples_matplotlib(
filepath.replace("$$", i),
filepath.replace("$$", str(i)),
[
# input_tensor,
input_pair[1], #label_colourmap