281 lines
13 KiB
Python
281 lines
13 KiB
Python
# OpenCV Webcam Script v0.5
|
||
# 创建人：曾逸夫
|
||
# 创建时间：2022-01-25
|
||
|
||
import cv2
|
||
import sys
|
||
import argparse
|
||
import time
|
||
from pathlib import Path
|
||
from datetime import datetime
|
||
from utils.ows_path import increment_path
|
||
from utils.hotkey import hotkey_judge
|
||
from utils.frame_opt import frame_opt
|
||
from utils.log import is_logSuffix, log_management, date_time_frames
|
||
from utils.args_yaml import argsYaml
|
||
from utils.compress import webcam_compress, is_compressFile
|
||
from utils.time_format import time_format
|
||
from utils.plot import csv2chart
|
||
from utils.fonts_opt import is_fonts
|
||
|
||
|
||
ROOT_PATH = sys.path[0] # 项目根目录
|
||
OWS_VERSION = 'OpenCV Webcam Script v0.5' # 项目名称与版本号
|
||
|
||
|
||
def parse_args(known=False):
|
||
parser = argparse.ArgumentParser(description='OpenCV Webcam Script v0.5')
|
||
parser.add_argument('--device', '-dev', default='0',
|
||
type=str, help='device index for webcam, 0 or rtsp')
|
||
parser.add_argument('--quit', '-q', default="q",
|
||
type=str, help='quit key for webcam')
|
||
parser.add_argument('--is_autoSaveFrame', '-isasf',
|
||
action='store_true', help='is auto save frame')
|
||
parser.add_argument('--is_handSaveFrame', '-ishsf',
|
||
action='store_true', help='is hand save frame')
|
||
parser.add_argument('--is_resizeFrame', '-isrf',
|
||
action='store_true', help='is resize frame')
|
||
parser.add_argument('--frame_saveDir', '-fsd',
|
||
default="./WebcamFrame", type=str, help='save frame dir')
|
||
parser.add_argument('--frame_dirName', '-fdn',
|
||
default="frames", type=str, help='save frame dir name')
|
||
parser.add_argument('--frame_nSave', '-fns', default=1,
|
||
type=int, help='n frames save a frame (auto save frame)')
|
||
parser.add_argument('--frame_capKey', '-fck', default="a",
|
||
type=str, help='frame capture key (hand save frame)')
|
||
parser.add_argument('--resize_frame', '-rf',
|
||
default="640,480", type=str, help='resize frame save')
|
||
parser.add_argument('--resizeRatio_frame', '-rrf',
|
||
default=1.0, type=float, help='resize ratio frame save')
|
||
parser.add_argument('--frame_namePrefix', '-fnp',
|
||
default="frame", type=str, help='frame name prefix')
|
||
parser.add_argument('--frame_saveStyle', '-fss',
|
||
default="jpg", type=str, help='frame save style')
|
||
parser.add_argument('--jpg_quality', '-jq',
|
||
default=95, type=int, help='frame save jpg quality (0-100) default 95')
|
||
parser.add_argument('--png_quality', '-pq',
|
||
default=3, type=int, help='frame save jpg quality (0-9) default 3')
|
||
parser.add_argument('--pause', '-p',
|
||
default="p", type=str, help='webcam pause')
|
||
parser.add_argument('--auto_frameNum', '-afn',
|
||
default=0, type=int, help='auto save number of frames')
|
||
|
||
# 日志
|
||
parser.add_argument('--logName', '-ln',
|
||
default="ows.log", type=str, help='log save name')
|
||
parser.add_argument('--logMode', '-lm',
|
||
default="a", type=str, help='log write mode')
|
||
# 压缩
|
||
parser.add_argument('--is_compress', '-isc',
|
||
action='store_true', help='is compress file')
|
||
parser.add_argument('--compressStyle', '-cs',
|
||
default="zip", type=str, help='compress style')
|
||
parser.add_argument('--is_autoCompressName', '-isacn',
|
||
action='store_true', help='is auto compress name')
|
||
parser.add_argument('--compressName', '-cn',
|
||
default="ows", type=str, help='compress save name')
|
||
parser.add_argument('--compressMode', '-cm',
|
||
default="w", type=str, help='compress save mode, tar w:gz')
|
||
args = parser.parse_known_args()[0] if known else parser.parse_args()
|
||
return args
|
||
|
||
|
||
# Webcam OpenCV
|
||
def webcam_opencv(device_index="0", # 设备号
|
||
quit_key="q", # 退出键
|
||
pause_key="p", # 暂停键
|
||
is_autoSaveFrame=False, # 自动保存帧
|
||
frame_saveDir="./WebcamFrame", # 帧保存路径
|
||
frame_dirName="frames", # 帧目录
|
||
frame_nSave=1, # 每隔n帧保存一次
|
||
auto_frameNum=0, # 自动保存最大帧数
|
||
is_handSaveFrame=False, # 手动保存帧
|
||
frame_capKey="a", # 设置帧捕获键
|
||
is_resizeFrame=False, # 重塑帧
|
||
resize_frame="640,480", # 自定义帧尺寸
|
||
resizeRatio_frame=1.0, # 自定义帧缩放比
|
||
frame_namePrefix="frame", # 自定义帧前缀
|
||
frame_saveStyle="jpg", # 帧保存类型
|
||
jpg_quality=95, # jpg质量系数
|
||
png_quality=3, # png质量系数
|
||
logName="ows.log", # 日志名称
|
||
logMode="a", # 日志模式
|
||
is_compress=False, # 压缩帧
|
||
compressStyle="zip", # 压缩类型
|
||
is_autoCompressName=False, # 自动命名压缩文件
|
||
compressName="ows", # 自定义压缩文件名称
|
||
compressMode="w"): # 压缩模式
|
||
|
||
keyList = [quit_key, frame_capKey, pause_key] # 快捷键列表
|
||
hotkey_judge(keyList) # 快捷键冲突判断
|
||
|
||
# 日志文件
|
||
is_logSuffix(logName) # 检测日志格式
|
||
logTime = f'{datetime.now():%Y-%m-%d %H:%M:%S}' # 日志时间
|
||
log_management(f'{logTime}\n', logName, logMode) # 记录日志时间
|
||
|
||
is_compressFile(compressStyle) # 检测压缩文件格式
|
||
|
||
dev_index = eval(device_index) if device_index.isnumeric(
|
||
) else device_index # 设备选择 (usb 0,1,2; rtsp)
|
||
cap = cv2.VideoCapture(dev_index) # 设备连接
|
||
is_capOpened = cap.isOpened() # 判断摄像头是否正常启动
|
||
|
||
if is_capOpened: # 设备连接成功
|
||
# ------------------程序开始------------------
|
||
s_time = time.time() # 起始时间
|
||
print(f'摄像头连接成功！')
|
||
print(f'-------------程序开始！-------------')
|
||
bufferSize = cap.get(cv2.CAP_PROP_BUFFERSIZE)
|
||
# cap.set(cv2.CAP_PROP_BUFFERSIZE,10) # 1-10
|
||
|
||
frame_width = cap.get(3) # 帧宽度
|
||
frame_height = cap.get(4) # 帧高度
|
||
fps = cap.get(5) # 帧率
|
||
print(f'宽度：{frame_width}, 高度：{frame_height}， FPS：{fps}， 缓存数：{bufferSize}')
|
||
|
||
frame_savePath = "" # 保存路径
|
||
if is_autoSaveFrame or is_handSaveFrame:
|
||
# 帧保存路径管理
|
||
frame_savePath = increment_path(
|
||
Path(f"{frame_saveDir}") / frame_dirName, exist_ok=False) # 增量运行
|
||
frame_savePath.mkdir(parents=True, exist_ok=True) # 创建目录
|
||
|
||
frame_num = 0 # 总帧数
|
||
frame_hand_num = 0 # 手动保存帧数
|
||
frame_n_num = 0 # 每隔n帧保存一次
|
||
|
||
while(is_capOpened):
|
||
wait_key = cv2.waitKey(20) & 0xFF # 键盘监听
|
||
_, frame = cap.read() # 捕获画面
|
||
frame_num += 1 # 帧计数
|
||
print(f'帧ID：{frame_num}') # 输出帧ID信息
|
||
cv2.imshow(OWS_VERSION, frame) # 显示画面
|
||
|
||
if (is_autoSaveFrame): # 自动保存
|
||
if (auto_frameNum > 0 and frame_num > auto_frameNum):
|
||
# 设置自动最大保存帧数
|
||
frame_num -= 1 # 修复帧数显示问题
|
||
break
|
||
if (frame_num % frame_nSave == 0): # 每隔n帧保存一次
|
||
frame_n_num += 1
|
||
frame_opt(frame, frame_savePath, frame_num, is_resizeFrame, resize_frame, resizeRatio_frame,
|
||
frame_namePrefix, frame_saveStyle, jpg_quality, png_quality)
|
||
elif (is_handSaveFrame): # 手动保存
|
||
if wait_key == ord(frame_capKey): # 保存键
|
||
frame_hand_num += 1 # 手动帧计数
|
||
frame_opt(frame, frame_savePath, frame_num, is_resizeFrame, resize_frame, resizeRatio_frame,
|
||
frame_namePrefix, frame_saveStyle, jpg_quality, png_quality)
|
||
|
||
if wait_key == ord(quit_key): # 退出 ord：字符转ASCII码
|
||
break
|
||
elif wait_key == ord(pause_key):
|
||
print(f'已暂停！按任意键继续。。。')
|
||
cv2.waitKey(0) # 暂停，按任意键继续
|
||
|
||
if (is_autoSaveFrame):
|
||
# 帧保存信息（自动版）
|
||
if (frame_n_num > 0):
|
||
frame_num = frame_n_num # 每隔n帧保存一次
|
||
frameSaveMsg = f'自动版：共计{frame_num}帧，已保存在：{frame_savePath}'
|
||
print(frameSaveMsg)
|
||
log_management(f'{frameSaveMsg}\n', logName, logMode) # 记录帧保存信息
|
||
date_time_frames(logTime, frame_num) # 记录时间与帧数
|
||
elif (is_handSaveFrame):
|
||
# 帧保存信息（手动版）
|
||
frameSaveMsg = f'手动版：共计{frame_hand_num}帧，已保存在：{frame_savePath}'
|
||
print(frameSaveMsg)
|
||
log_management(f'{frameSaveMsg}\n', logName, logMode) # 记录帧保存信息
|
||
date_time_frames(logTime, frame_hand_num) # 记录时间与帧数
|
||
else:
|
||
date_time_frames(logTime, 0) # 记录非帧保存状态
|
||
|
||
cap.release() # 释放缓存资源
|
||
cv2.destroyAllWindows() # 删除所有窗口
|
||
|
||
# ------------------程序结束------------------
|
||
print(f'-------------程序结束！-------------')
|
||
e_time = time.time() # 终止时间
|
||
total_time = e_time - s_time # 程序用时
|
||
outTimeMsg = f'用时：{time_format(total_time)}' # 格式化时间格式，便于观察
|
||
print(outTimeMsg)
|
||
log_management(f'{outTimeMsg}\n', logName, logMode) # 记录用时
|
||
|
||
# ------------------压缩文件------------------
|
||
if (is_compress and (is_autoSaveFrame or is_handSaveFrame)):
|
||
# 压缩信息
|
||
compress_msg = webcam_compress(compressStyle, is_autoCompressName,
|
||
compressName, frame_savePath, compressMode)
|
||
log_management(f'{compress_msg}\n', logName, logMode) # 记录用时
|
||
|
||
# ------------------创建chart------------------
|
||
is_fonts(f'{ROOT_PATH}/fonts') # 检查字体文件
|
||
csv2chart("./date_time_frames.csv") # 创建日期-帧数图
|
||
|
||
else: # 连接设备失败
|
||
print(f'摄像头连接异常！')
|
||
|
||
|
||
def main(args):
|
||
device_index = args.device
|
||
quit_key = args.quit
|
||
is_autoSaveFrame = args.is_autoSaveFrame
|
||
is_handSaveFrame = args.is_handSaveFrame
|
||
frame_saveDir = args.frame_saveDir
|
||
frame_dirName = args.frame_dirName
|
||
frame_nSave = args.frame_nSave
|
||
frame_capKey = args.frame_capKey
|
||
resize_frame = args.resize_frame
|
||
is_resizeFrame = args.is_resizeFrame
|
||
resizeRatio_frame = args.resizeRatio_frame
|
||
frame_namePrefix = args.frame_namePrefix
|
||
frame_saveStyle = args.frame_saveStyle
|
||
jpg_quality = args.jpg_quality
|
||
png_quality = args.png_quality
|
||
pause_key = args.pause
|
||
auto_frameNum = args.auto_frameNum
|
||
|
||
# 日志
|
||
logName = args.logName
|
||
logMode = args.logMode
|
||
|
||
# 压缩
|
||
is_compress = args.is_compress
|
||
compressStyle = args.compressStyle
|
||
is_autoCompressName = args.is_autoCompressName
|
||
compressName = args.compressName
|
||
compressMode = args.compressMode
|
||
|
||
argsYaml(args) # 脚本参数
|
||
|
||
# 调用webcam opencv
|
||
webcam_opencv(device_index,
|
||
quit_key,
|
||
pause_key,
|
||
is_autoSaveFrame,
|
||
frame_saveDir,
|
||
frame_dirName,
|
||
frame_nSave,
|
||
auto_frameNum,
|
||
is_handSaveFrame,
|
||
frame_capKey,
|
||
is_resizeFrame,
|
||
resize_frame,
|
||
resizeRatio_frame,
|
||
frame_namePrefix,
|
||
frame_saveStyle,
|
||
jpg_quality,
|
||
png_quality,
|
||
logName,
|
||
logMode,
|
||
is_compress,
|
||
compressStyle,
|
||
is_autoCompressName,
|
||
compressName,
|
||
compressMode)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
args = parse_args()
|
||
main(args)
|