python视频转图片帧，图片帧转线稿，线稿合并为视频

最新推荐文章于 2023-10-04 21:10:58 发布

黑小慕

最新推荐文章于 2023-10-04 21:10:58 发布

阅读量644

点赞数 1

分类专栏： python 文章标签： python 音视频 opencv

本文链接：https://blog.csdn.net/m0_67790374/article/details/126844777

版权

python 专栏收录该内容

41 篇文章 16 订阅

订阅专栏

原视频：一人之下

一人之下

视频转图片帧

图片帧转线稿

线稿合并为视频

线稿视频

python程序

import os
import sys
import cv2
import imageio.v2 as imageio
import numpy as np
from PIL import Image, ImageFilter, ImageOps

def video_to_frame(video):#视频转图片帧
    path=Image.open(video)
    if path is None:
        print('没有找到该文件，程序结束！')
        sys.exit()
    if os.path.exists('./帧/')==False:
        os.mkdir('./帧/')
    cap=cv2.VideoCapture(video)
    k=0
    while True:
        if cap.grab():
            k+=1
            flag,frame=cap.retrieve()
            if flag:
                new=f'./帧/{k}.jpg'
                print(f'正在生成图片帧：{k}.jpg')
                cv2.imencode('.jpg', frame)[1].tofile(new)
        else:
            break

def frame_to_line():#图片帧转线稿
    img_array=os.listdir('./帧/')
    if os.path.exists('./线稿')==False:
        os.mkdir('./线稿')
    for k in range(1,len(img_array)+1):
        img=Image.open(f'./帧/{k}.jpg')
        img1=img.convert('L')
        img2=img1.copy()
        img2=ImageOps.invert(img2)
        for i in range(25):
            img2=img2.filter(ImageFilter.BLUR)
        width,height=img1.size
        for x in range(width):
            for y in range(height):
                a=img1.getpixel((x,y))
                b=img2.getpixel((x,y))
                img1.putpixel((x,y),min(int(a*255/(256-b*1)),255))
        img1.save(f'./线稿/{k}.jpg')
        print(f'正在转换为线稿：{k}.jpg')

def line_to_video():#线稿合并为视频
    img_array=os.listdir('./线稿/')
    img_array.sort(key=lambda x:int(x[:-4]))
    pic=Image.open('./线稿/'+img_array[0])
    video=cv2.VideoWriter('线稿视频.mp4',cv2.VideoWriter_fourcc('m','p','4','v'),20, (pic.size[0],pic.size[1]))
    for i in range(1,len(img_array)+1):
        img=imageio.imread('./线稿/'+img_array[i-1])
        img2=cv2.cvtColor(img,cv2.COLOR_GRAY2BGR)
        video.write(img2)
    video.release()
    print('线稿视频.mp4已生成！')

if __name__=='__main__':
    video=input('请输入视频文件路径(如：一人之下.mp4)：')
    video_to_frame(video)
    frame_to_line()
    line_to_video()