added model file existence check

human_detection/yolov7_skeleton/sort.py

@@ -3,7 +3,7 @@
 import os
 import numpy as np
 import matplotlib
-matplotlib.use('TkAgg')
+# matplotlib.use('TkAgg')
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 from skimage import io

human_detection/yolov7_skeleton/yolov7.py

@@ -10,6 +10,7 @@
 from utils.general import check_img_size, non_max_suppression, scale_coords, set_logging, increment_path
 from utils.torch_utils import select_device, TracedModel
 import sort
+import os 
 
 """Function to draw bounding boxes"""
 def draw_boxes(img, bbox, identities=None, categories=None, confidences=None, names=None, colors=None, thickness=2, hide_bounding_box=False,hide_labels=False):
@@ -64,6 +65,9 @@ def __init__(self,
         self.use_half_precision = self.device.type != 'cpu'  # enable half precision if on GPU (only supported on CUDA)
 
         # Load model
+        if not os.path.exists(weights_file):
+            raise FileNotFoundError(f"The file '{weights_file}' does not exist.")
+
         self.model = attempt_load(weights_file, map_location=self.device)  # load FP32 model
         self.stride = int(self.model.stride.max())  # model stride, which is the step size or the number of units the sliding window moves when performing operations like convolution or pooling
         self.imgsize = check_img_size(img_size, s=self.stride)  # check img_size