Update visualize.py

mrcnn/visualize.py

@@ -189,7 +189,7 @@ def display_instances(image, boxes, masks, class_ids, class_names,
     if save_img:
         #choose = round(number_saved_images * random.random())                                                                                     # save processed image
         if (counter_1 == number_saved_images):
-            plt.savefig(os.path.join(save_dir, img_name + '.jpg'),bbox_inches='tight', pad_inches=False, orientation= 'landscape')   
+            plt.savefig(os.path.join(save_dir, img_name),bbox_inches='tight', pad_inches=False, orientation= 'landscape')   
     
 plt.close()