有没有办法只显示视频中只有检测到对象边界框的帧

Question

我正在使用视频在测试数据上测试我的自定义 yolov5s 对象检测模型。然而，并不是视频中的每一帧都有一个有效的对象被检测到。

[ 当推理测试在视频上运行时，下图显示模型何时找到具有可以检测和分类的对象的帧

如您所见，帧 4401 已检测到让行交通标志，但帧 4399 和 4400 未检测到任何内容。我的问题是如何处理 4500 帧视频并仅显示检测到对象的帧。我已将帧提取为 JPEGS，但只想要有效的 JPEGS 来评估测试性能？更好的是，是否可以将相关帧中的数据提取到文本文件中？

如果有人能帮助我，这将是一个很大的帮助和节省时间，在此先感谢。

the new error

Answer 1

from shutil import move
import os

txt_dir = "txtfiles"
images_folder = "images_folder"
move_des = "move_imgs_path"

for file in os.listdir(txt_dir):
    if file.endswith('txt'):
          frame_name = file.split('.')[0]+ ".jpeg"
          print(frame_name)
          move(os.path.join(images_folder, frame_name), move_des)

有没有办法只显示视频中只有检测到对象边界框的帧

Is there a way to only display frames from a video that only have detected object bounding boxes

python

python-3.x

pytorch

yolo

yolov5