描述：將一個視頻流按幀數(shù)截取大量的圖片

用途：ai的數(shù)據(jù)集制作，得到大量的圖片，之后將其打標(biāo)簽

更改的地方

1.default--間隔的幀數(shù)? ?2.input/output--輸入視頻的路徑、存放截取圖片的路徑 (將路徑放入后面的‘ ’中即可)前面加r可表示絕對路徑 eg:

1 args = parser.parse_args(['--input',r'f:\data_video\img_4395.mov','--output',r'f:data_rgb_pic\7video'])

直接上代碼

1 import cv2

2 import argparse

3 import os

4 def parse_args():

5 """

6 parse input arguments

7 """

8 parser = argparse.argumentparser(description='process pic')

9 parser.add_argument('--input', help='video to process', dest='input', default=none, type=str)

10 parser.add_argument('--output', help='pic to store', dest='output', default=none, type=str)

11 #default為間隔多少幀截取一張圖片

12 parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=100, type=int)

13 #input為輸入視頻的路徑 ，output為輸出存放圖片的路徑

14 args = parser.parse_args(['--input','','--output',''])

15 return args

16

17 def process_video(i_video, o_video, num):

18 cap = cv2.videocapture(i_video)

19 num_frame = cap.get(cv2.cap_prop_frame_count)

20 expand_name = '.jpg'

21 if not cap.isopened():

22 print("please check the path.")

23 cnt = 0

24 count = 0

25 while 1:

26 ret, frame = cap.read()

27 cnt += 1

28 # how

29 # many

30 # frame

31 # to

32 # cut

33 if cnt % num == 0:

34 count += 1

35 cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame)

36

37 if not ret:

38 break

39

40 if __name__ == '__main__':

41 args = parse_args()

42 if not os.path.exists(args.output):

43 os.makedirs(args.output)

44 print('called with args:')

45 print(args)

46 process_video(args.input, args.output, args.skip_frame)

參考

如您對本文有疑問或者有任何想說的，請點擊進行留言回復(fù)，萬千網(wǎng)友為您解惑！

總結(jié)

以上是生活随笔為你收集整理的python截取视频图像_python工具-将视频按帧截取图片（附代码）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。