图片变文件 视频提取图片

Author: budaLi

视频截取图片/0_51_Dresses_wearingdress_51_377_0.png

@@ -0,0 +1,41 @@
+# @Time    : 2020/5/28 14:25
+# @Author  : Libuda
+# @FileName: 图片变脚本.py
+# @Software: PyCharm
+# -*- coding: utf-8 -*-
+"""以python模块形式存储、使用二进制文件"""
+import os
+import base64
+from io import BytesIO
+
+
+def bin2module(bin_file, py_file=None):
+    """二进制文件转存为python模块
+    bin_file  - 二进制文件名
+    py_file   - 生成的模块文件名，默认使用二进制文件名，仅更改后缀名
+    """
+    fpath, fname = os.path.split(bin_file)
+    fn, ext = os.path.splitext(fname)
+    if not py_file:
+        py_file = os.path.join(fpath, '%s.py' % fn)
+    with open(bin_file, 'rb') as fp:
+        content = fp.read()
+    content = base64.b64encode(content)
+    content = content.decode('utf8')
+    with open(py_file, 'w') as fp:
+        fp.write('content = """%s"""\n\n' % content)
+
+
+if __name__ == '__main__':
+    """测试代码"""
+    # 将图像文件转存为img_demo.py
+    bin2module('0_51_Dresses_wearingdress_51_377_0.png', 'demo.py')
+    # 导入刚刚生成的demo模块
+    import demo
+    # 用pillow打开图像，验证demo模块的get_fp()：返回二进制的IO对象（类文件对象）
+    from PIL import Image
+
+    im = Image.open(demo.get_fp())
+    im.show()
+    # 保存为本地文件，验证demo模块的save()：保存文件
+    demo.save('demo_save.png')

视频截取图片/38f98723f3525f0518537c1d36677cf1.mp4

@@ -0,0 +1,58 @@
+# @Time    : 2020/5/28 14:20
+# @Author  : Libuda
+# @FileName: 视频变图片帧.py
+# @Software: PyCharm
+
+import argparse
+import os
+
+import cv2
+
+
+def parse_args():
+    """
+    Parse input arguments
+    """
+    parser = argparse.ArgumentParser(description='Process pic')
+    parser.add_argument('--input', help='video to process', dest='input',
+                        default="38f98723f3525f0518537c1d36677cf1.mp4", type=str)
+    parser.add_argument('--output', help='pic to store', dest='output', default="./imgs", type=str)
+    # default为间隔多少帧截取一张图片
+    parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=1,
+                        type=int)  # 此处可更改提取帧的间隔
+    args = parser.parse_args()  # 此处添加路径，input为输入视频的路径 ，output为输出存放图片的路径
+    return args
+
+
+def process_video(i_video, o_video, num):
+    cap = cv2.VideoCapture(i_video)
+    num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT)
+    expand_name = '.png'
+    if not cap.isOpened():
+        print("Please check the path.")
+    cnt = 0
+    count = 0
+    while 1:
+        ret, frame = cap.read()
+        cnt += 1
+        # how
+        # many
+        # frame
+        # to
+        # cut
+        if cnt % num == 0:
+            count += 1
+            print(frame)
+            frame = cv2.resize(frame, (128, 128))
+            cv2.imwrite(os.path.join(o_video, "shimei" + str(count) + expand_name), frame)
+        if not ret:
+            break
+
+
+if __name__ == '__main__':
+    args = parse_args()
+    if not os.path.exists(args.output):
+        os.makedirs(args.output)
+    print('Called with args:')
+    print(args)
+    process_video(args.input, args.output, args.skip_frame)