Python - 截取指定幀數間隔指定大小的視頻，並保存圖片到指定位置

Python - 截取指定幀數間隔指定大小的視頻，並保存圖片到指定位置

環境：Python3.6

依賴庫：shutil、os、cv2

視頻如下：

代碼如下：

# -*- coding: utf-8 -*-

import shutil
import os
import cv2

os.chdir(os.path.split(os.path.realpath(__file__))[0])
print(os.getcwd())

dirs_path = os.getcwd()                         # 設置路徑
    
dirs = os.listdir(dirs_path)                    # 獲取指定路徑下的文件
for video_name in dirs:                             # 循環讀取路徑下的文件並篩選輸出
	if os.path.splitext(video_name)[1] == ".mp4":   # 篩選mp4文件
		print(video_name)                            # 輸出所有的mp4文件
		break

output_img_path = './output1/'
if not os.path.exists(output_img_path):
    os.mkdir(output_img_path)
else:
    # 刪除原有目錄
    shutil.rmtree(output_img_path)
    # 創建一個新目錄
    os.mkdir(output_img_path)

cap = cv2.VideoCapture(video_name)   
#videoFPS=cap.get(cv2.CAP_PROP_FPS)
#print (videoFPS)

cnt = 1

print ("start")
   
while(1):
    ret, frame = cap.read()
    # show a frame
    if ret is True:
        print(frame.shape)
        cnt=cnt+1
        print(cnt)
        if cnt % 10 == 0:
#            cropped = frame[1075:128, 119:1397]  # 裁剪座標爲[y0:y1, x0:x1]
            cropped = frame[120:750, 260:1660]  # 裁剪座標爲[y0:y1, x0:x1]
            output_name = "%s%s%s"%(output_img_path,str(int(cnt/10)),'.jpg')
            cv2.imwrite(output_name, cropped)
    else:
        break
print("finished")        

cap.release()
cv2.destroyAllWindows()

備註：

爲方便代碼執行，直接將文件可以放置視頻目錄下，自動檢測MP4文件，
其中cropped = frame[120:750, 260:1660] # 裁剪座標爲[y0:y1, x0:x1]可以設置裁剪的圖片大小
if cnt % 10 == 0:爲間隔10幀截取一次圖片
output_img_path = './output1/'爲圖片保存目錄

Result

去掉了水印以及修改了相應的大小

代碼下載地址

https://download.csdn.net/download/tanghong1996/11011868