bugfix: image read error while drawing FN when there is no detection …

…on an image
Franksongsong · Mar 15, 2020 · 2ca0ff2 · 2ca0ff2
1 parent 838d9eb
commit 2ca0ff2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -741,6 +741,9 @@ def draw_plot_func(dictionary, n_classes, window_title, plot_title, x_label, out
     img_id = tmp_file[tmp_file.find(start)+len(start):tmp_file.rfind('_ground_truth.json')]
     img_cumulative_path = output_files_path + "/images/" + img_id + ".jpg"
     img = cv2.imread(img_cumulative_path)
+    if img is None:
+        img_path = IMG_PATH + '/' + img_id + ".jpg"
+        img = cv2.imread(img_path)
     # draw false negatives
     for obj in ground_truth_data:
         if not obj['used']: