菜橙 2022-07-19 09:36 采纳率: 75%
浏览 1752
已结题

yolov5检测视频如何保存每帧结果?

请问YOLOV5 检测视频时如何将每一帧的图像都保存下来,需要在当帧检测完成后完成画框就保存,而不是结束后对视频进行切分。
目前使用过cv2.imwrite,但是只能保存有目标的帧。修改部分代码如下:

            vidcut_dir = str(save_dir) + '/vidcut'
            if not os.path.exists(vidcut_dir):
                os.makedirs(vidcut_dir)
            vidcut_path = vidcut_dir + '\\' + str(p.stem) + ('' if dataset.mode == 'image' else f'_{frame}')  

                # Write results
                for *xyxy, conf, cls in reversed(det):
                    if save_txt:  # Write to file
                        xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist()  # normalized xywh
                        line = (cls, *xywh, conf) if opt.save_conf else (cls, *xywh)  # label format
                        with open(txt_path + '.txt', 'a') as f:
                            f.write(('%g ' * len(line)).rstrip() % line + '\n')
                    if save_img or view_img:  # Add bbox to image
                        label = f'{names[int(cls)]} {conf:.2f}'
                        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)
                        #修改部分-----------------------------------------------------------------
                        a0 = (int(xyxy[0].item()) + int(xyxy[2].item())) / 2
                        if a0 != 0:
                            cv2.imwrite(vidcut_path + f'{p.stem}.jpg', im0)
                        else:
                            im1 = cv2.imread('no.jpg',1)
                            cv2.imwrite(vidcut_path + f'{p.stem}.jpg', im1)

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2022-07-19 10:27
    关注

    img


    你在这里面存肯定是有目标才会进入for循环,把存图的代码提到for循环外面去即可

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录

    报告相同问题?

    问题事件

    • 系统已结题 7月27日
    • 已采纳回答 7月19日
    • 创建了问题 7月19日

    悬赏问题

    • ¥20 C语言字符串不区分大小写字典排序相关问题
    • ¥15 关于#python#的问题:我希望通过逆向技术爬取1688搜索页下滑加载的数据
    • ¥15 学习C++过程中遇到的问题
    • ¥15 关于Linux的终端里,模拟实现一个带口令保护的屏保程序遇到的输入输出的问题!(语言-c语言)
    • ¥15 学习C++过程中遇到的问题
    • ¥15 请问,这个嵌入式Linux系统怎么分析,crc检验区域在哪
    • ¥15 二分类改为多分类问题
    • ¥15 Unity微信小游戏上调用ReadPixels()方法报错
    • ¥15 如何通过求后验分布求得样本中属于两种物种其中一种的概率?
    • ¥15 q从常量变成sin函数,怎么改写python代码?