| 1234567891011121314151617181920 |
- from supervision.assets import download_assets, VideoAssets
- import supervision as sv
- import cv2
- import os
- from inference import get_model
- from ultralytics import YOLO
- print(sv.__version__)
- HOME = os.getcwd()
- download_assets(VideoAssets.VEHICLES)
- VIDEO_PATH = VideoAssets.VEHICLES.value
- sv.VideoInfo.from_video_path(video_path=VIDEO_PATH)
- frame_generator = sv.get_video_frames_generator(source_path=VIDEO_PATH)
- frame = next(iter(frame_generator))
- sv.plot_image(image=frame, size=(4, 4))
- RESULT_VIDEO_PATH = f"{HOME}/videos/vehicle-counting-result.mp4"
|
