python记录9-CFANZ编程社区

基于PyCharm实现Flask分镜

1.进行视频切帧

定义切帧函数genFrame

def genFrame():
    v_path='static/itzy-output.mp4'
    image_save='static/hash'
 
    if not(os.path.exists(image_save)):
        print(image_save)
        os.mkdir(image_save)
 
    cap=cv2.VideoCapture(v_path)
    fc=cap.get(cv2.CAP_PROP_FRAME_COUNT)
    print(fc)
    _, img1 = cap.read()
    cv2.imwrite('static/hash/image{}.jpg'.format(0), img1)
    print(int(fc))
    for i in range(248):
        _, img2 = cap.read()
        hash1 = aHash(img1)
        hash2 = aHash(img2)
        n = cmpHash(hash1, hash2)
        #print('均值哈希算法相似度：', n)
        if (n<0.6):
            cv2.imwrite('static/hash/image{}.jpg'.format(i),img2)
            img1=img2

查看切帧结果

 
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Flask分镜</title>
</head>
<body>
视频分镜
<br>
<video width="640" height="480" controls autoplay>
  <source src="static/video.mp4" type="video/mp4">
  <object data="static/video.mp4" width="640" height="480">
    <embed width="640" height="480" src="static/itzy-output.mp4">
  </object>
</video>
<br>
帧数：{{framecount}}<br>
{% for i in range(framecount) %}
<img height="40" src="{{pic1}}{{i}}.jpg" />
{% endfor %}
 
</body>
</html>

利用hash均值进行分镜提取

# 均值哈希算法
def aHash(img):
    # 转换为灰度图
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    s = 0
    hash_str = ''
    for i in range(8):
        for j in range(8):
            s = s + gray[i, j]
    # 求平均灰度
    avg = s / 64
    for i in range(8):
        for j in range(8):
            if gray[i, j] > avg:
                hash_str = hash_str + '1'
            else:
                hash_str = hash_str + '0'
    return hash_str
 
# Hash值对比
def cmpHash(hash1, hash2):
    n = 0
    # hash长度不同则返回-1代表传参出错
    if len(hash1)!=len(hash2):
        return -1
    # 遍历判断
    for i in range(len(hash1)):
        # 不相等则n计数+1，n最终为相似度
        if hash1[i] != hash2[i]:
            n = n + 1
    sim=1-n/64
    return sim

@app.route('/')
def shot():
    genFrame()
    path='static/hash'
    filename = os.listdir(path)
    framecount=len(filename)
    filename.sort(key=lambda x: int(x[5:-4]))
    print(filename)
    print(type(filename))
 
    return render_template('shot.html', filename=filename, framecount=framecount)
if '__main__'==__name__:
    app.run(port='5008')