out = cv2.VideoWriter("output_video.mp4", cv2.VideoWriter_fourcc(*"mp4v"), 30, (frame.shape[1], frame.shape[0])) while video.isOpened(): ret, frame = video.read() if not ret: break results = model(frame) for box in results.boxes: x1, y1, x2, y2, conf, cls = box.xyxy[0], box.conf, box.cls label = f"{results.names[int(cls)]} {conf:.2f}" cv2.rectangle(frame, (int(x1), int(y1)), (int(x2), int(y2)), (0, 255, 0), 2) cv2.putText(frame, label, (int(x1), int(y1) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2) out.write(frame) video.release() out.release() out = cv2.VideoWriter("output_video.mp4", cv2.VideoWriter_fourcc(*"mp4v"), 30, (frame.shape[1], frame.shape[0])) while video.isOpened(): ret, frame = video.read() if not ret: break results = model(frame) for box in results.boxes: x1, y1, x2, y2, conf, cls = box.xyxy[0], box.conf, box.cls label = f"{results.names[int(cls)]} {conf:.2f}" cv2.rectangle(frame, (int(x1), int(y1)), (int(x2), int(y2)), (0, 255, 0), 2) cv2.putText(frame, label, (int(x1), int(y1) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2) out.write(frame) video.release() out.release()
top of page
bottom of page