frames dir name

2022-01-08 09:54:23 +08:00 · 2022-01-08 09:54:23 +08:00 · e0964f1763
parent fee46975a9
commit e0964f1763
3 changed files with 18 additions and 160 deletions
--- a/opencv_webcam.py
+++ b/opencv_webcam.py
@ -1,6 +1,6 @@
-# OpenCV Webcam Script v0.2
+# OpenCV Webcam Script v0.3
 # 创建人：曾逸夫
-# 创建时间：2022-01-02
+# 创建时间：2022-01-08

 import cv2
 from pathlib import Path
@ -15,9 +15,9 @@ from utils.frame_opt import frame_opt


 def parse_args(known=False):
-    parser = argparse.ArgumentParser(description='OpenCV Webcam Script v0.2')
-    parser.add_argument('--device', '-dev', default=0,
-                        type=int, help='device index for webcam')
+    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',
@ -28,6 +28,8 @@ def parse_args(known=False):
                        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",
@ -54,15 +56,18 @@ def parse_args(known=False):

 # Webcam OpenCV
 def webcam_opencv(device_index=0, quit_key="q", pause_key="p", is_autoSaveFrame=False, frame_saveDir="./WebcamFrame",
-                  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):
+                  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()  # 起始时间
-    cap = cv2.VideoCapture(device_index)  # 连接设备
+    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)  # 帧率
@ -75,11 +80,11 @@ def webcam_opencv(device_index=0, quit_key="q", pause_key="p", is_autoSaveFrame=
        if is_autoSaveFrame or is_handSaveFrame:
            # 帧保存路径管理
            frame_savePath = increment_path(
-                Path(f"{frame_saveDir}") / "frames", exist_ok=False)  # 增量运行
+                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.2'  # 版本号
+        ows_version = 'OpenCV Webcam Script v0.3'  # 版本号
        while(1):
            wait_key = cv2.waitKey(20) & 0xFF  # 键盘监听
            ret, frame = cap.read()  # 捕获画面
@ -124,6 +129,7 @@ def main(args):
    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
@ -137,7 +143,7 @@ def main(args):
    auto_frameNum = args.auto_frameNum

    # 调用webcam opencv
-    webcam_opencv(device_index, quit_key, pause_key, is_autoSaveFrame, frame_saveDir, frame_nSave, auto_frameNum,
+    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)

--- a/opencv_webcam_v03.py
+++ b/opencv_webcam_v03.py
@ -1,149 +0,0 @@
-# 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_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_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}") / "frames", 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_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_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)
--- a/v03.md
+++ b/v03.md
@ -8,6 +8,7 @@

 ### 功能更新
 - 加入rtsp功能，比如：rtsp://username:password@xxx.xxx.xxx
+- 加入自定义帧目录名称


 ### BUG修复