Update test_h5image.py

test_h5image.py

@@ -20,7 +20,7 @@ legend_resized = tf.image.resize(legend_patch, (256,256))
 if legend_resized.dtype == tf.float32:
     legend_resized = (legend_resized * 255).numpy().astype(np.uint8)
 
-print("legend_resized.shape", legend_resized.shape, "unique values", np.unique(legend_patch), np.unique(legend_resized))
+print("legend", legend, "legend_resized.shape", legend_resized.shape, "unique values", np.unique(legend_patch))
 
 # Create figure
 fig, ax = plt.subplots(figsize=(3, 3)) # Adjust the size as needed