Python - витяг та збереження відеокадрів

Тому я дотримувався цього підручника, але, здається, він нічого не робить. Просто нічого. Він чекає кілька секунд і закриває програму. Що не так з цим кодом?

import cv2
vidcap = cv2.VideoCapture('Compton.mp4')
success,image = vidcap.read()
count = 0
success = True
while success:
  success,image = vidcap.read()
  cv2.imwrite("frame%d.jpg" % count, image)     # save frame as JPEG file
  if cv2.waitKey(10) == 27:                     # exit if Escape is hit
      break
  count += 1

Також у коментарях сказано, що це обмежує рамки до 1000? Чому?

EDIT: Я спробував зробити success = Trueперший, але це не допомогло. Він створив лише одне зображення, яке було 0 байт.

Від сюди завантажити це відео таким чином , ми маємо той же відеофайл для тестування. Переконайтеся, що цей mp4 файл знаходиться в одному каталозі вашого python-коду. Потім також переконайтеся, що запустіть інтерпретатор python з тієї самої директорії.

Потім змініть код, ров, waitKeyякий витрачає час також без вікна, він не може фіксувати події клавіатури. Також ми друкуємо successзначення, щоб переконатися, що воно успішно читає кадри.

import cv2
vidcap = cv2.VideoCapture('big_buck_bunny_720p_5mb.mp4')
success,image = vidcap.read()
count = 0
while success:
  cv2.imwrite("frame%d.jpg" % count, image)     # save frame as JPEG file      
  success,image = vidcap.read()
  print('Read a new frame: ', success)
  count += 1

Як це йде?

Поширити це питання (& відповідь від @ user2700065) на дещо інші випадки, якщо хтось не хоче витягти кожен кадр, але хоче витягнути кадр щосекунди. Таким чином, 1-хвилинне відео дасть 60 кадрів (зображень).

import sys
import argparse

import cv2
print(cv2.__version__)

def extractImages(pathIn, pathOut):
    count = 0
    vidcap = cv2.VideoCapture(pathIn)
    success,image = vidcap.read()
    success = True
    while success:
        vidcap.set(cv2.CAP_PROP_POS_MSEC,(count*1000))    # added this line 
        success,image = vidcap.read()
        print ('Read a new frame: ', success)
        cv2.imwrite( pathOut + "\\frame%d.jpg" % count, image)     # save frame as JPEG file
        count = count + 1

if __name__=="__main__":
    a = argparse.ArgumentParser()
    a.add_argument("--pathIn", help="path to video")
    a.add_argument("--pathOut", help="path to images")
    args = a.parse_args()
    print(args)
    extractImages(args.pathIn, args.pathOut)

Це налаштування з попередньої відповіді на python 3.x від @GShocked, я б опублікував це до коментаря, але не маю достатньої репутації

import sys
import argparse

import cv2
print(cv2.__version__)

def extractImages(pathIn, pathOut):
    vidcap = cv2.VideoCapture(pathIn)
    success,image = vidcap.read()
    count = 0
    success = True
    while success:
      success,image = vidcap.read()
      print ('Read a new frame: ', success)
      cv2.imwrite( pathOut + "\\frame%d.jpg" % count, image)     # save frame as JPEG file
      count += 1

if __name__=="__main__":
    print("aba")
    a = argparse.ArgumentParser()
    a.add_argument("--pathIn", help="path to video")
    a.add_argument("--pathOut", help="path to images")
    args = a.parse_args()
    print(args)
    extractImages(args.pathIn, args.pathOut)

Ця функція перетворить більшість форматів відео в кількість кадрів, які є у відео. Він працює на Python3зOpenCV 3+

import cv2
import time
import os

def video_to_frames(input_loc, output_loc):
    """Function to extract frames from input video file
    and save them as separate frames in an output directory.
    Args:
        input_loc: Input video file.
        output_loc: Output directory to save the frames.
    Returns:
        None
    """
    try:
        os.mkdir(output_loc)
    except OSError:
        pass
    # Log the time
    time_start = time.time()
    # Start capturing the feed
    cap = cv2.VideoCapture(input_loc)
    # Find the number of frames
    video_length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) - 1
    print ("Number of frames: ", video_length)
    count = 0
    print ("Converting video..\n")
    # Start converting the video
    while cap.isOpened():
        # Extract the frame
        ret, frame = cap.read()
        # Write the results back to output location.
        cv2.imwrite(output_loc + "/%#05d.jpg" % (count+1), frame)
        count = count + 1
        # If there are no more frames left
        if (count > (video_length-1)):
            # Log the time again
            time_end = time.time()
            # Release the feed
            cap.release()
            # Print stats
            print ("Done extracting frames.\n%d frames extracted" % count)
            print ("It took %d seconds forconversion." % (time_end-time_start))
            break

if __name__=="__main__":

    input_loc = '/path/to/video/00009.MTS'
    output_loc = '/path/to/output/frames/'
    video_to_frames(input_loc, output_loc)

Він підтримує .mtsі звичайні файли, такі як .mp4і .avi. Пробував і тестував на .mtsфайлах. Працює як Чарівність.

Після багатьох досліджень, як перетворити кадри у відео, я створив цю функцію, сподіваюся, це допоможе. Для цього нам потрібен opencv:

import cv2
import numpy as np
import os

def frames_to_video(inputpath,outputpath,fps):
   image_array = []
   files = [f for f in os.listdir(inputpath) if isfile(join(inputpath, f))]
   files.sort(key = lambda x: int(x[5:-4]))
   for i in range(len(files)):
       img = cv2.imread(inputpath + files[i])
       size =  (img.shape[1],img.shape[0])
       img = cv2.resize(img,size)
       image_array.append(img)
   fourcc = cv2.VideoWriter_fourcc('D', 'I', 'V', 'X')
   out = cv2.VideoWriter(outputpath,fourcc, fps, size)
   for i in range(len(image_array)):
       out.write(image_array[i])
   out.release()


inputpath = 'folder path'
outpath =  'video file path/video.mp4'
fps = 29
frames_to_video(inputpath,outpath,fps)

змінити значення кадрів в секунду (кадрів в секунду), шлях вхідної папки та шлях до папки відповідно до ваших місцевих локацій

Попередні відповіді втратили перший кадр. І буде приємно зберігати зображення в папці.

# create a folder to store extracted images
import os
folder = 'test'  
os.mkdir(folder)
# use opencv to do the job
import cv2
print(cv2.__version__)  # my version is 3.1.0
vidcap = cv2.VideoCapture('test_video.mp4')
count = 0
while True:
    success,image = vidcap.read()
    if not success:
        break
    cv2.imwrite(os.path.join(folder,"frame{:d}.jpg".format(count)), image)     # save frame as JPEG file
    count += 1
print("{} images are extacted in {}.".format(count,folder))

До речі, ви можете перевірити рамку щура e за допомогою VLC. Перейдіть до Windows -> інформація про медіа -> деталі кодека

Цей код витягує кадри з відео та зберігає кадри у форматі .jpg

import cv2
import numpy as np
import os

# set video file path of input video with name and extension
vid = cv2.VideoCapture('VideoPath')


if not os.path.exists('images'):
    os.makedirs('images')

#for frame identity
index = 0
while(True):
    # Extract images
    ret, frame = vid.read()
    # end of frames
    if not ret: 
        break
    # Saves images
    name = './images/frame' + str(index) + '.jpg'
    print ('Creating...' + name)
    cv2.imwrite(name, frame)

    # next frame
    index += 1

Я використовую Python через програмне забезпечення Spyder Anaconda. Використовуючи оригінальний код, перелічений у питанні цієї теми від @Gshocked, код не працює (python не прочитає файл mp4). Тому я завантажив OpenCV 3.2 і скопіював "opencv_ffmpeg320.dll" і "opencv_ffmpeg320_64.dll" з папки "bin". Я вставив обидва ці файли dll у папку "Dlls" Анаконда.

Анаконда також має папку "pckgs" ... Я скопіював і вставив всю папку "OpenCV 3.2", яку я завантажив у папку Anaconda "pckgs".

Нарешті, у Anaconda є папка "Бібліотека", в якій є підпапка "bin". Я вставив файли "opencv_ffmpeg320.dll" і "opencv_ffmpeg320_64.dll" у цю папку.

Після закриття та перезавантаження Spyder код спрацював. Я не впевнений, який із трьох методів працював, і я лінивий, щоб повернутися назад і зрозуміти це. Але це працює так, ура!

Ця функція витягує зображення з відео з 1 кадром в секунду, В ДОДАТКІ він визначає останній кадр і зупиняє читання також:

import cv2
import numpy as np

def extract_image_one_fps(video_source_path):

    vidcap = cv2.VideoCapture(video_source_path)
    count = 0
    success = True
    while success:
      vidcap.set(cv2.CAP_PROP_POS_MSEC,(count*1000))      
      success,image = vidcap.read()

      ## Stop when last frame is identified
      image_last = cv2.imread("frame{}.png".format(count-1))
      if np.array_equal(image,image_last):
          break

      cv2.imwrite("frame%d.png" % count, image)     # save frame as PNG file
      print '{}.sec reading a new frame: {} '.format(count,success)
      count += 1

Наступний сценарій буде витягувати кадри кожні півсекунди всіх відео в папці. (Працює на python 3.7)

import cv2
import os
listing = os.listdir(r'D:/Images/AllVideos')
count=1
for vid in listing:
    vid = r"D:/Images/AllVideos/"+vid
    vidcap = cv2.VideoCapture(vid)
    def getFrame(sec):
        vidcap.set(cv2.CAP_PROP_POS_MSEC,sec*1000)
        hasFrames,image = vidcap.read()
        if hasFrames:
            cv2.imwrite("D:/Images/Frames/image"+str(count)+".jpg", image) # Save frame as JPG file
        return hasFrames
    sec = 0
    frameRate = 0.5 # Change this number to 1 for each 1 second
    
    success = getFrame(sec)
    while success:
        count = count + 1
        sec = sec + frameRate
        sec = round(sec, 2)
        success = getFrame(sec)