if save_vid: if vid_path != save_path: # new video print("save=", save_path) vid_path = save_path if isinstance(vid_writer, cv2.VideoWriter): vid_writer.release() # release previous video writer if vid_cap: # video fps = vid_cap.get(cv2.CAP_PROP_FPS) w = int(vid_cap.get(cv2.CAP_PROP_FRAME_WIDTH)) 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' vid_writer = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h)) vid_writer.write(im0)改成保存结果为图片

如果要将结果保存为图片，可以使用以下代码：

if save_img:
    cv2.imwrite(save_path, im0)

这将使用cv2.imwrite将图像写入指定的路径。请注意，这将只保存单帧图像，而不是整个视频。若要保存整个视频，需要更改代码以为每个帧创建单独的图像，并将其保存为图像序列。