仅用作个人记录
import cv2
import os
def video_thread(fname):
inputFileName = fname.replace('\\','/')
outputFileName=inputFileName.replace('input','outputs').replace('.265','')+'/'
cap=cv2.VideoCapture(inputFileName)
isOpened=cap.isOpened()
print(isOpened)
i=0
while isOpened:
i+=1
ret,frame=cap.read()
print(ret)
if ret==True:
cv2.namedWindow('image', 0)
cv2.imshow('image',frame)
cv2.waitKey(1)
if not os.path.exists(outputFileName):
os.makedirs(outputFileName)
cv2.imwrite(outputFileName + 'frame_' + str(i) + '.jpg', frame)
else:
cap.release()
cv2.destroyAllWindows()
return 0
cap.release()
cv2.destroyAllWindows()
def train():
path = "input"
for root, dirs, files in os.walk(path):
for file in files:
filename = root + '/' + file
video_thread(filename)
if __name__ == '__main__':
train()