diff --git a/plot_image.py b/plot_image.py index ad355fdcc415499fdbd0f4bfda99da2521c68ee8..a636ff40c60b739996041e0f1cc799aeefb7c4fd 100644 --- a/plot_image.py +++ b/plot_image.py @@ -46,7 +46,12 @@ def show_images(images, cols=1, titles=None): """ # default embody image for the background - default_img = plt.imread("./dummy_600.png") + + try: + background = True + default_img = plt.imread("./dummy_600.png") + except FileNotFoundError: + background = False # get a copy of the gray color map my_cmap = copy.copy(plt.cm.get_cmap('gray')) @@ -67,7 +72,8 @@ def show_images(images, cols=1, titles=None): plt.imshow(image, cmap=my_cmap) # draw default background image with transparency on top of the points - plt.imshow(default_img, extent=[0, 200, 600, 0], alpha=0.33) + if background: + plt.imshow(default_img, extent=[0, 200, 600, 0], alpha=0.33) a.set_title(title)