Update app.py

app.py

@@ -34,7 +34,7 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         img.save(img_path)
         input_path = img_path
 
-        output_path, df, frame_counts_df = run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True)
+        output_path, df, frame_counts_df = run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True, hide_labels = True)
     elif vid_path is not None:
         vid_name = 'output.mp4'
 
@@ -74,9 +74,9 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
             output_path, df, frame_counts_df = run_deepsort(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', draw_trails=True)
         elif tracking_algorithm == 'strong_sort':
             device_strongsort = torch.device('cuda:0')
-            output_path, df, frame_counts_df = run_strongsort(yolo_weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device=device_strongsort, strong_sort_weights = "osnet_x0_25_msmt17.pt", hide_conf= True)
+            output_path, df, frame_counts_df = run_strongsort(yolo_weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device=device_strongsort, strong_sort_weights = "osnet_x0_25_msmt17.pt", hide_conf= True,hide_labels = True)
         else: 
-            output_path, df, frame_counts_df =  run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True)
+            output_path, df, frame_counts_df =  run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True, hide_labels = True)
         # Assuming output_path is the path to the output file
     _, output_extension = os.path.splitext(output_path)
     palette = {"Bus": "red", "Bike": "blue", "Car": "green", "Pedestrian": "yellow", "Truck": "purple"}
@@ -91,9 +91,9 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         sns.barplot(ax=ax, data=df, x='label', y='count', palette=palette, hue='label')
 
         # Customizations
-        ax.set_title('Count of Labels', fontsize=20, pad=20)  # Increase padding for the title
-        ax.set_xlabel('Label', fontsize=16)  # Increase font size
-        ax.set_ylabel('Count', fontsize=16)  # Increase font size
+        ax.set_title('Number of Objects', fontsize=20, pad=20)  # Increase padding for the title
+        ax.set_xlabel('Object Class', fontsize=16)  # Increase font size
+        ax.set_ylabel('Object Count', fontsize=16)  # Increase font size
         ax.tick_params(axis='x', rotation=45, labelsize=12)  # Increase label size and rotate x-axis labels for better readability
         ax.tick_params(axis='y', labelsize=12)  # Increase label size for y-axis
         sns.despine()  # Remove the top and right spines from plot
@@ -116,9 +116,9 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
             #df_label = frame_counts_df[frame_counts_df['label'] == label]
         sns.lineplot(ax = ax, data = frame_counts_df,  x = 'frame', y = 'count', hue = 'label', palette=palette,linewidth=2.5)
 
-        ax.set_title('Count of Labels over Frames', fontsize=20, pad=20)  # Increase padding for the title
-        ax.set_xlabel('Frame', fontsize=16)  # Increase font size
-        ax.set_ylabel('Count', fontsize=16)  # Increase font size
+        ax.set_title('Number of Objects over Seconds', fontsize=20, pad=20)  # Increase padding for the title
+        ax.set_xlabel('Second', fontsize=16)  # Increase font size
+        ax.set_ylabel('Object Count', fontsize=16)  # Increase font size
         ax.tick_params(axis='x', labelsize=12)  # Increase label size for x-axis
         ax.tick_params(axis='y', labelsize=12)  # Increase label size for y-axis