fixing Cartucho#106

main.py

@@ -732,24 +732,25 @@ def draw_plot_func(dictionary, n_classes, window_title, plot_title, x_label, out
 """
  Draw false negatives
 """
-pink = (203,192,255)
-for tmp_file in gt_files:
-    ground_truth_data = json.load(open(tmp_file))
-    #print(ground_truth_data)
-    # get name of corresponding image
-    start = TEMP_FILES_PATH + '/'
-    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']:
-            bbgt = [ int(round(float(x))) for x in obj["bbox"].split() ]
-            cv2.rectangle(img,(bbgt[0],bbgt[1]),(bbgt[2],bbgt[3]),pink,2)
-    cv2.imwrite(img_cumulative_path, img)
+if show_animation:
+    pink = (203,192,255)
+    for tmp_file in gt_files:
+        ground_truth_data = json.load(open(tmp_file))
+        #print(ground_truth_data)
+        # get name of corresponding image
+        start = TEMP_FILES_PATH + '/'
+        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']:
+                bbgt = [ int(round(float(x))) for x in obj["bbox"].split() ]
+                cv2.rectangle(img,(bbgt[0],bbgt[1]),(bbgt[2],bbgt[3]),pink,2)
+        cv2.imwrite(img_cumulative_path, img)
 
 # remove the temp_files directory
 shutil.rmtree(TEMP_FILES_PATH)