【自動取冰機(jī)】如何從視頻中提取所有包含冰冰的畫面?

經(jīng)常看到B站網(wǎng)友們上傳一些冰冰的圖片,甚是羨慕。
久而久之,想起了一句古話:與其臨淵羨魚,不如退而結(jié)網(wǎng)。于是,我決定自制一臺自動取冰機(jī),可以從視頻中自動提取所有包含冰冰畫面的圖片。
下面是自動取冰機(jī)的大致制作流程,且當(dāng)自娛自樂吧。
步驟1:下載幾個關(guān)于冰冰的視頻,以備后用
這里以央視影音為例,搜索關(guān)于冰冰的視頻,然后點擊下載即可。下載完成后,在“我的下載”那里可以看到下載到的視頻。


步驟2:使用opencv-python將視頻素材拆解成圖片,以備后用


根據(jù)上面的代碼,運(yùn)行后,我們就可以將準(zhǔn)備好的視頻素材拆解為圖片了,這些圖片作為后面的訓(xùn)練數(shù)據(jù)使用。
步驟3:從上一步驟的圖片素材中篩選出包含冰冰的圖片,然后給它打標(biāo)簽
很遺憾,在自動取冰機(jī)制作出來之前,這一步的圖片篩選只能靠自己的眼睛和雙手了,篩選的速度有多快,取決于你的手速有多快。
篩選好了之后,我們使用LabelImage工具給這些圖片打標(biāo)簽。

這一步,需要注意的是,打標(biāo)簽的圖片需要按VOC的格式進(jìn)行組織(其實就是把圖片和xml文件放到指定名字的目錄)。

步驟4:將打好標(biāo)簽的數(shù)據(jù)集拿到百度的AIStudio上進(jìn)行訓(xùn)練


步驟5:將AIStudio上訓(xùn)練好的模型下載到本地

步驟6:使用訓(xùn)練好的模型對視頻文件進(jìn)行探測,提取冰元素

至此,自動取冰機(jī)已經(jīng)制作完成了。
我們只需要下載好要提取冰元素的視頻原料,然后在代碼中設(shè)置文件路徑,接著運(yùn)行即可。

最后,程序運(yùn)行完畢后,我們就能在輸出目錄看到“自動取冰機(jī)”從視頻原料中提取出來的所有冰元素了。

