Add detect.py GIF video inference (#6410)

* Add detect.py GIF video inference

* Cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-01-24 13:11:11 -08:00
1 parent 8efe97719c
commit 482af479c0
2 files changed
+3 -3

No files matched your search

+1 -1
View File
@@ -199,7 +199,7 @@ def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s)
h = int(vid_cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
else: # stream
fps, w, h = 30, im0.shape[1], im0.shape[0]
save_path += '.mp4'
save_path = str(Path(save_path).with_suffix('.mp4')) # force *.mp4 suffix on results videos
vid_writer[i] = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h))
vid_writer[i].write(im0)