【自动取冰机】如何从视频中提取所有包含冰冰的画面？

经常看到B站网友们上传一些冰冰的图片，甚是羡慕。

久而久之，想起了一句古话：与其临渊羡鱼，不如退而结网。于是，我决定自制一台自动取冰机，可以从视频中自动提取所有包含冰冰画面的图片。

下面是自动取冰机的大致制作流程，且当自娱自乐吧。

步骤1：下载几个关于冰冰的视频，以备后用

这里以央视影音为例，搜索关于冰冰的视频，然后点击下载即可。下载完成后，在“我的下载”那里可以看到下载到的视频。

步骤2：使用opencv-python将视频素材拆解成图片，以备后用

根据上面的代码，运行后，我们就可以将准备好的视频素材拆解为图片了，这些图片作为后面的训练数据使用。

步骤3：从上一步骤的图片素材中筛选出包含冰冰的图片，然后给它打标签

很遗憾，在自动取冰机制作出来之前，这一步的图片筛选只能靠自己的眼睛和双手了，筛选的速度有多快，取决于你的手速有多快。

筛选好了之后，我们使用LabelImage工具给这些图片打标签。

这一步，需要注意的是，打标签的图片需要按VOC的格式进行组织（其实就是把图片和xml文件放到指定名字的目录）。

步骤4：将打好标签的数据集拿到百度的AIStudio上进行训练

步骤5：将AIStudio上训练好的模型下载到本地

步骤6：使用训练好的模型对视频文件进行探测，提取冰元素

至此，自动取冰机已经制作完成了。

我们只需要下载好要提取冰元素的视频原料，然后在代码中设置文件路径，接着运行即可。

最后，程序运行完毕后，我们就能在输出目录看到“自动取冰机”从视频原料中提取出来的所有冰元素了。