Abhinav Deshpande commited on
Bug: Model Path fix
Browse files
Backend/BrandRecognition/Dynamic/Brand_Count_Vid.py
CHANGED
|
@@ -24,7 +24,7 @@ def smooth_box(box_history):
|
|
| 24 |
return np.mean(box_history, axis=0)
|
| 25 |
|
| 26 |
def process_video(input_path, output_path):
|
| 27 |
-
model = YOLO('kitkat_s.pt')
|
| 28 |
cap = cv2.VideoCapture(input_path)
|
| 29 |
|
| 30 |
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
|
|
|
| 24 |
return np.mean(box_history, axis=0)
|
| 25 |
|
| 26 |
def process_video(input_path, output_path):
|
| 27 |
+
model = YOLO('Weights/kitkat_s.pt')
|
| 28 |
cap = cv2.VideoCapture(input_path)
|
| 29 |
|
| 30 |
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|