目录
可以直接安装:pip install ultralytics
推理可视化,加nms效果更好,imageio保存mp4
import os
from glob import glob
import torch
import cv2
import torchvision
from ultralytics import YOLO
import imageio
def load_video_frames(video_path, v_len=10000):
frames = []
count = 0
if os.path.isdir(video_path):
files = glob(video_path + '/*.png') + glob(video_path + '/*.jpg')
for file in files:
img = cv2.imread(file)
frames.append(cv