Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ for i, url in enumerate(file_urls):
|
|
29 |
f"image_{i}.jpg"
|
30 |
)
|
31 |
|
32 |
-
|
33 |
path = [['image_0.jpg'], ['image_1.jpg']]
|
34 |
video_path = [['video.mp4']]
|
35 |
|
@@ -76,7 +76,7 @@ def show_preds_video(video_path):
|
|
76 |
frame_copy,
|
77 |
(int(det[0]), int(det[1])),
|
78 |
(int(det[2]), int(det[3])),
|
79 |
-
|
80 |
thickness=2,
|
81 |
lineType=cv2.LINE_AA
|
82 |
)
|
|
|
29 |
f"image_{i}.jpg"
|
30 |
)
|
31 |
|
32 |
+
model = YOLO('best.pt')
|
33 |
path = [['image_0.jpg'], ['image_1.jpg']]
|
34 |
video_path = [['video.mp4']]
|
35 |
|
|
|
76 |
frame_copy,
|
77 |
(int(det[0]), int(det[1])),
|
78 |
(int(det[2]), int(det[3])),
|
79 |
+
color=(0, 0, 255),
|
80 |
thickness=2,
|
81 |
lineType=cv2.LINE_AA
|
82 |
)
|