BPG压缩生成残差图像

东山阿强

于 2021-01-06 10:09:21 发布

阅读量982

点赞数

CC 4.0 BY-SA版权

分类专栏：编解码文章标签： cv

本文链接：https://blog.csdn.net/weixin_43306331/article/details/112258827

编解码专栏收录该内容

2 篇文章

订阅专栏

import os
from PIL import Image
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
import numpy as np 
os.system("mkdir bpg")
os.system("mkdir png")
os.system("mkdir rgb_residual")
path = os.listdir('./')
i = 1
for fn in path:
    if fn.endswith(".png"):
        fd1 = os.path.join('./', fn)
        fd2 = os.path.join('./png', fn)
        sh1 = "bpgenc -b 8 -q 51 " + fd1 + " -o ./bpg/" + fn[:-4] +".bpg"
        sh2 = "bpgdec ./bpg/" + fn[:-4] +".bpg" + " -o ./png/" + fn[:-4] +".png"
        sh3 = "rm -rf " + "./bpg/" + fn[:-4] +".bpg"
        sh4 = "rm -rf " + fd2
        os.system(sh1)
        os.system(sh2)
        src1 = Image.open(fd1).convert('RGB')
        src2 = Image.open(fd2).convert('RGB')
        rgb1 = np.array(src1)
        rgb2 = np.array(src2)
        # yuv = yuv + 128.0
        filename = os.path.join('./rgb_residual', fn)
        #print(yuv)
        #rgb = rgb + 128
        rgb = rgb2 - rgb1 + 128
        cv2.imwrite(filename, rgb)
        i += 1
        if i%10 == 0:
            print(i)
        os.system(sh3)
        os.system(sh4)