exp updte
This commit is contained in:
13339479676
parent
c10d045a7c
commit
d910955df8
|
|
@ -0,0 +1,153 @@
|
|||
# OpenCV Webcam Script v0.3
|
||||
# 创建人：曾逸夫
|
||||
# 创建时间：2022-01-08
|
||||
|
||||
import cv2
|
||||
from pathlib import Path
|
||||
import argparse
|
||||
import time
|
||||
import glob
|
||||
import os
|
||||
import sys
|
||||
from utils.ows_path import increment_path
|
||||
from utils.hotkeyjudge import hotkey_judge
|
||||
from utils.frame_opt import frame_opt
|
||||
|
||||
|
||||
def parse_args(known=False):
|
||||
parser = argparse.ArgumentParser(description='OpenCV Webcam Script v0.3')
|
||||
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=int, nargs='+', 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')
|
||||
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, 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, png_quality=3):
|
||||
|
||||
keyList = [quit_key, frame_capKey, pause_key] # 快捷键列表
|
||||
hotkey_judge(keyList) # 快捷键冲突判断
|
||||
|
||||
# ------------------程序开始------------------
|
||||
s_time = time.time() # 起始时间
|
||||
dev_index = eval(device_index) if device_index.isnumeric(
|
||||
) else device_index # 设备选择 (usb 0,1,2; rtsp)
|
||||
cap = cv2.VideoCapture(dev_index) # 连接设备
|
||||
frame_width = cap.get(3) # 宽度
|
||||
frame_height = cap.get(4) # 高度
|
||||
fps = cap.get(5) # 帧率
|
||||
print(f'宽度：{frame_width}, 高度：{frame_height}， FPS：{fps}')
|
||||
success, _ = cap.read() # 读取设备
|
||||
|
||||
if success: # 连接设备成功
|
||||
print(f'摄像头连接成功！')
|
||||
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 # 帧数
|
||||
ows_version = 'OpenCV Webcam Script v0.3' # 版本号
|
||||
while(1):
|
||||
wait_key = cv2.waitKey(20) & 0xFF # 键盘监听
|
||||
ret, frame = cap.read() # 捕获画面
|
||||
frame_num += 1 # 帧计数
|
||||
print(f'帧ID：{frame_num}') # 输出信息
|
||||
cv2.imshow(ows_version, frame) # 显示画面
|
||||
|
||||
if (is_autoSaveFrame): # 自动保存
|
||||
if (auto_frameNum > 0 and frame_num > auto_frameNum):
|
||||
# 设置自动最大保存帧数
|
||||
break
|
||||
if (frame_num % frame_nSave == 0): # 每隔n帧保存一次
|
||||
frame_opt(frame, frame_savePath, frame_num, is_resizeFrame, resize_frame, resizeRatio_frame,
|
||||
frame_namePrefix, frame_saveStyle, jpg_quality, png_quality)
|
||||
if (is_handSaveFrame): # 手动保存
|
||||
if wait_key == ord(frame_capKey): # 保存键
|
||||
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
|
||||
if wait_key == ord(pause_key):
|
||||
print(f'已暂停！按任意键继续。。。')
|
||||
cv2.waitKey(0) # 暂停，按任意键继续
|
||||
|
||||
if (frame_savePath != ""):
|
||||
print(f'共计{frame_num}帧，已保存在：{frame_savePath}')
|
||||
|
||||
cap.release() # 释放缓存资源
|
||||
cv2.destroyAllWindows() # 删除所有窗口
|
||||
else: # 连接设备失败
|
||||
print(f'摄像头连接异常！')
|
||||
# ------------------程序结束------------------
|
||||
print(f'程序结束！')
|
||||
e_time = time.time() # 终止时间
|
||||
print(f'用时：{round((e_time - s_time), 3)}秒, {round((e_time - s_time)/60, 3)}分, {round((e_time - s_time)/3600, 3)}小时')
|
||||
|
||||
|
||||
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
|
||||
|
||||
# 调用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)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
args = parse_args()
|
||||
main(args)
|
||||
|
|
@ -112,9 +112,9 @@ def webcam_opencv(device_index=0, quit_key="q", pause_key="p", is_autoSaveFrame=
|
|||
|
||||
if (frame_savePath != ""):
|
||||
print(f'共计{frame_num}帧，已保存在：{frame_savePath}')
|
||||
# 释放缓存资源
|
||||
cap.release()
|
||||
cv2.destroyAllWindows()
|
||||
|
||||
cap.release() # 释放缓存资源
|
||||
cv2.destroyAllWindows() # 删除所有窗口
|
||||
else: # 连接设备失败
|
||||
print(f'摄像头连接异常！')
|
||||
# ------------------程序结束------------------
|
||||
|
|
|
|||
Loading…
Reference in New Issue