show_tile_images

utils/show_tile_images.py

@@ -1,8 +1,17 @@
-from utils.fig2img import fig2img
 import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 from typing import List
+import io
+
+def fig2img(fig: plt.Figure):
+  """Convert a Matplotlib figure to a PIL Image and return it"""
+  plt.close()
+  buf = io.BytesIO()
+  fig.savefig(buf)
+  buf.seek(0)
+  img = Image.open(buf)
+  return img
 
 def show_tile_images(
     images: List[np.ndarray | Image.Image],