# -*- coding: utf-8 -*-
|
|
|
|
|
#将视频分给为一张一张的图片
|
__author__ = '上海色宁'
|
|
import os
|
import sys
|
import cv2
|
import numpy as np
|
import uuid
|
import getopt
|
|
|
def start_check_argv_and_split(argv):
|
"""
|
通过 getopt模块 来识别参数demo, http://blog.csdn.net/ouyang_peng/
|
"""
|
input_video = ""
|
clip_rect = ""
|
dest_dir = ""
|
|
try:
|
"""
|
options, args = getopt.getopt(args, shortopts, longopts=[])
|
|
参数args：一般是sys.argv[1:]。过滤掉sys.argv[0]，它是执行脚本的名字，不算做命令行参数。
|
参数shortopts：短格式分析串。例如："hp:i:"，h后面没有冒号，表示后面不带参数；p和i后面带有冒号，表示后面带参数。
|
参数longopts：长格式分析串列表。例如：["help", "ip=", "port="]，help后面没有等号，表示后面不带参数；ip和port后面带冒号，表示后面带参数。
|
返回值options是以元组为元素的列表，每个元组的形式为：(选项串, 附加参数)，如：('-i', '192.168.0.1')
|
返回值args是个列表，其中的元素是那些不含'-'或'--'的参数。
|
"""
|
opts, args = getopt.getopt(argv, "hi:c:d:", ["help", "input_video=", "clip_rect=","dest_dir="])
|
except getopt.GetoptError:
|
print('Error: 0_split_video_into_image.py -i <input_video> -c <clip_rect> -d <dest_dir>')
|
print(' or: 0_split_video_into_image.py --input_video=<video.mp4> --clip_rect=<[175:235,275:420]> --dest_dir=<split_image>')
|
sys.exit(2)
|
|
# 处理 返回值options是以元组为元素的列表。
|
if len(opts) != 3 :
|
print('Error: Parameters not enough!')
|
print('Usage: 0_split_video_into_image.py -i <input_video> -c <clip_rect> -d <dest_dir>')
|
print(' or: 0_split_video_into_image.py --input_video=<video.mp4> --clip_rect=<175:235,275:420> --dest_dir=<split_image>')
|
sys.exit()
|
for opt, arg in opts:
|
if opt in ("-h", "--h","-help","--help"):
|
print('Usage: 0_split_video_into_image.py -i <input_video> -c <clip_rect> -d <dest_dir>')
|
print(' or: 0_split_video_into_image.py --input_video=<video.mp4> --clip_rect=<175:235,275:420> --dest_dir=<split_image>')
|
sys.exit()
|
elif opt in ("-i", "--input_video"):
|
input_video = arg
|
elif opt in ("-c", "--clip_rect"):
|
clip_rect = arg
|
elif opt in ("-d", "--dest_dir"):
|
dest_dir = arg
|
print('input_video:', input_video)
|
print('clip_rect:', clip_rect )
|
print(' . !!! we are assuming that this paramter is correct.')
|
print('dest_dir:', dest_dir)
|
|
if (not os.path.isfile(input_video)):
|
print( input_video+ " is not exist!!!")
|
sys.exit()
|
if not os.path.exists(dest_dir):
|
print( dest_dir+ " is not exist, we are going to creat it !!!")
|
os.makedirs(dest_dir)
|
|
# 打印 返回值args列表，即其中的元素是那些不含'-'或'--'的参数。
|
#for i in range(0, len(args)):
|
# print('参数 %s 为：%s' % (i + 1, args[i]))
|
|
split_into_image(input_video,clip_rect,dest_dir)
|
|
|
def split_into_image(input_video,clip_rect,dest_dir):
|
|
cap = cv2.VideoCapture(input_video)
|
rows = clip_rect.split(',')[0] # 行
|
cols = clip_rect.split(',')[1] # 列
|
rows_start = int(rows.split(':')[0])
|
rows_end = int(rows.split(':')[1])
|
cols_start = int(cols.split(':')[0])
|
cols_end = int(cols.split(':')[1])
|
|
totol_cols = (cols_end-cols_start)
|
col_length_1 = int(totol_cols*0.212352113)
|
col_length_2 = int(totol_cols*0.212352113)
|
col_length_3_4 = int(totol_cols*0.212352113)
|
|
|
col_start_2 = cols_start+int(totol_cols*0.265690141)
|
col_start_3 = cols_start+int(totol_cols*0.485746479)
|
col_start_4 = cols_start+int(totol_cols*0.75943662)
|
|
i = 0
|
while True:
|
if i % 10 == 0:
|
ret, frame = cap.read()
|
#cv2.imshow("capture", frame[rows_start:rows_end,cols_start:cols_end])
|
cv2.imwrite(dest_dir+"/"+str(uuid.uuid4())+".jpg", frame[rows_start:rows_end,cols_start:cols_start+col_length_1])
|
cv2.imwrite(dest_dir+"/"+str(uuid.uuid4())+".jpg", frame[rows_start:rows_end,col_start_2:col_start_2+col_length_2])
|
cv2.imwrite(dest_dir+"/"+str(uuid.uuid4())+".jpg", frame[rows_start:rows_end,col_start_3:col_start_3+col_length_3_4])
|
cv2.imwrite(dest_dir+"/"+str(uuid.uuid4())+".jpg", frame[rows_start:rows_end,col_start_4:col_start_4+col_length_3_4])
|
if cv2.waitKey(1) & 0xFF == ord('q'):
|
break
|
i += 1
|
cap.release()
|
cv2.destroyAllWindows()
|
|
|
#程序入口
|
if __name__ == '__main__':
|
start_check_argv_and_split(sys.argv[1:])
|
