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

Question

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

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

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

如果有人可以帮助我，这将是一个很大的帮助和节省时间，非常感谢提前。

新的错误

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)