【Python】動画をフレームごとの画像にして出力するスクリプトを作った話

はじまり#

135ml

動画から画像を切り出したいな。

リサちゃん

はい、目玉焼きができたよ～。

ツールの紹介#

さて、今回使う機能は、ExtractImage()という関数になります。

スクリプトを起動すると、ファイルを選択するダイアログが表示されます。今回は、このビデオを画像にします。

選択すると、画像を保存するフォルダの名前を入力します。※日本語（SJIS）は使えません！

そして、画像化が始まると順次保存されたファイルの名前を出力します。

画像化が終わると、保存したフォルダを確認するように促すメッセージが表示されて、スクリプトが終了します。

先程の動画があるフォルダを見ると、確かに先程入力したフォルダがあります。

そのフォルダの中を確認すると、確かに画像が入っています。おおよそ１秒毎に切り取っています。

スクリプトの紹介#

こちらが今回のスクリプトです。そこまで複雑なスクリプトではないと思います。

1
def ExtractImage(video_name):
2
    '''
3
    video_name    : String absolutely path of selected file
4
    extracted_dir : String absolutely path of directory has selected file
5
    frame_count   : Integer number of frame of selected video
6
    fps           : Integer number of fps of selected video
7
    num_of_image  : Integer number of extracted images from video
8
    '''
9
    # Error Handling
10
    if video_name == '':
11
        print("ERROR: No file is selected.")
12
        sys.exit(0)
13

14
    extracted_dir = DirEditor.MakeDirectory(video_name)
15

16
    # Error Handling
17
    basefilename_without_ext = os.path.splitext(os.path.basename(__file__))[0]
18
    if InputController.CheckWhetherSjisExists(extracted_dir, basefilename_without_ext) == True:
19
        sys.exit(0)
20

21
    cap = cv2.VideoCapture(video_name)
22
    # width
23
    width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
24
    # height
25
    height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
26
    # number of frame
27
    frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)
28
    # fps
29
    fps = cap.get(cv2.CAP_PROP_FPS)
30
    print("movie's width: ", width, ", height: ", height)
31

32
    for num_of_image in range(1, int(frame_count), int(fps)):
33
        # set a property, where frame in this case, in the VideoCapture
34
        cap.set(cv2.CAP_PROP_POS_FRAMES, num_of_image)
35
        # read the next video frame to extract image file ("cap.read()[1]" is arrays of image's pixels.)
36
        cv2.imwrite("{dirname}{sep}image{num:0=3}.jpg".format(dirname=extracted_dir,sep=sep,num=int((num_of_image-1)/int(fps))), cap.read()[1])
37
        # cv2.imwrite("{video_dir}image{:0=3}.jpg".format(int((num_of_image-1)/int(fps))), cap.read()[1])
38
        print("saved: image{num:0=3}.jpg".format(num=int((num_of_image-1)/int(fps))))
39
    print('Check directory "{dirname}"'.format(dirname=extracted_dir))
40
    cap.release()