好了,接下来我要做一个实际的深度学习图像分割的小项目,项目内容是从一堆拍摄海面的图片中将白浪花分割出来,这个项目的分割只对白浪花感兴趣,所以最后应该是01分割,非黑即白。目前收到800G左右的数据,视频格式,每段大约50分钟。首先要做的就是从这些视频中把每一帧的图片导出来,变成图片。
这里首先有一个小插曲,我发现我的视频格式一开始虽然是mp4,但是用普通的播放器播放不了,这就很诡异。于是我下载了完美解码,果然可以播放了。不过还是很诡异,于是我用格式工厂无损的转换成mp4格式,800G的数据也跑了整整一天时间,我真佛了。
下面贴上使用python 用opencv接口把视频逐帧转化为图片的程序,当然matlab也能干这个事儿,虽然我matlab比python熟,但是以后都用python编的,所以干脆用python了。
代码语言:javascript复制import cv2
import os
def save_img():
video_path = 'C:/Users/86151/Desktop/test'
videos = os.listdir(video_path)
for video_name in videos:
file_name = video_name.split('.')[0]
folder_name = video_path file_name
os.makedirs(folder_name, exist_ok=True)
vc = cv2.VideoCapture(video_path '/' video_name)
c=0
rval=vc.isOpened()
while rval:
c = c 1
rval, frame = vc.read()
pic_path = folder_name '/'
if rval:
cv2.imwrite(pic_path str(c) '.png', frame)
cv2.waitKey(1)
else:
break
vc.release()
print('save_success')
print(folder_name)
save_img()
需要特别注意:
代码语言:javascript复制video_path = 'C:/Users/86151/Desktop/test'