def bpg_measure(input_image_p,
bpp,
precise=False,
save_output_as_png=None,
tmp_dir=None):
"""
:return (PSNR, SSIM, MS-SSIM, actual_bpp)
"""
input_image_root_p, ext = os.path.splitext(input_image_p)
assert ext == '.png', 'Expected PNG to convert to BMP, got {}'.format(
input_image_p)
output_image_bpg_p, actual_bpp = _bpg_compress_to_bpp(input_image_p,
target_bpp=bpp,
precise=precise)
output_image_bpg_png_p = decode_bpg_to_png(output_image_bpg_p)
os.remove(output_image_bpg_p) # don't need that anymore
_, msssim, _ = compare_imgs.compare(input_image_p,
output_image_bpg_png_p,
calc_ssim=False,
calc_msssim=True,
calc_psnr=False)
if save_output_as_png:
os.rename(output_image_bpg_png_p, save_output_as_png)
else:
os.remove(output_image_bpg_png_p)
return msssim, actual_bpp
def balle_measure_over_interval(input_image_p,
fout,
grid,
model='mbt2018-mean-mse-'):
# input_image_root_p, ext = os.path.splitext(input_image_p)
# assert ext == '.png', 'Expected PNG to convert to BMP, got {}'.format(input_image_p)
output_image_p = input_image_p + '.tmp.tfci'
for q in map(str, grid):
with remove_file_after(
balle_compress(model, input_image_p, output_image_p,
q)) as output_image_p:
bpp = bpp_of_balle_image(input_image_p, output_image_p)
with remove_file_after(
decode_balle_to_png(output_image_p)) as out_png_p:
im_in = np.array(Image.open(input_image_p))
im_out = np.array(Image.open(out_png_p))
ssim, msssim, psnr = compare_imgs.compare(im_in,
im_out,
calc_ssim=True,
calc_msssim=True,
calc_psnr=True)
fout_str = _append_to_measures_f(fout, int(q), bpp, ssim,
msssim, psnr)
print(fout_str, end='\r')
print()
def jp_measure_over_interval(input_image_p, fout, q_grid):
input_image_root_p, ext = os.path.splitext(input_image_p)
assert ext == '.png'
for q in q_grid:
actual_bpp, output_image_jp_p = jp_compress(input_image_p, q)
im_in = np.array(Image.open(input_image_p))
im_out = np.array(Image.open(output_image_jp_p))
ssim, msssim, psnr = compare_imgs.compare(im_in,
im_out,
calc_ssim=True,
calc_msssim=True,
calc_psnr=True)
fout_str = _append_to_measures_f(fout, q, actual_bpp, ssim, msssim,
psnr)
print(fout_str, end='\r')
os.remove(output_image_jp_p)
print()
def webp_measure_over_interval(input_image_p, fout, q_grid):
input_image_root_p, ext = os.path.splitext(input_image_p)
assert ext == '.png'
for q in map(int, q_grid):
out_p, bpp = _webp_compress(input_image_p, q)
with remove_file_after(out_p) as out_p:
with remove_file_after(_decode_webp_to_png(out_p)) as out_png_p:
im_in = np.array(Image.open(input_image_p))
im_out = np.array(Image.open(out_png_p))
ssim, msssim, psnr = compare_imgs.compare(im_in,
im_out,
calc_ssim=True,
calc_msssim=True,
calc_psnr=True)
fout_str = _append_to_measures_f(fout, q, bpp, ssim, msssim,
psnr)
print(fout_str, end='\r')
print()
def bpg_measure_over_interval(input_image_p, fout, grid, m=8):
input_image_root_p, ext = os.path.splitext(input_image_p)
assert ext == '.png', 'Expected PNG to convert to BMP, got {}'.format(
input_image_p)
for q in map(int, grid):
with remove_file_after(bpg_compress(input_image_p, q)) as out_p:
bpp = bpp_of_bpg_image(out_p)
with remove_file_after(decode_bpg_to_png(out_p)) as out_png_p:
im_in = np.array(Image.open(input_image_p))
im_out = np.array(Image.open(out_png_p))
ssim, msssim, psnr = compare_imgs.compare(im_in,
im_out,
calc_ssim=True,
calc_msssim=True,
calc_psnr=True)
fout_str = _append_to_measures_f(fout, q, bpp, ssim, msssim,
psnr)
print(fout_str, end='\r')
print()
def jp2k_measure_over_interval(input_image_p, fout, bpp_grid):
input_image_root_p, ext = os.path.splitext(input_image_p)
assert ext == '.png', 'Expected PNG to convert to BMP, got {}'.format(
input_image_p)
for bpp in bpp_grid:
output_image_j2_p, actual_bpp = jp2k_compress(input_image_p, bpp)
with remove_file_after(
decode_jp2k_to_png(output_image_j2_p)) as out_png_p:
im_in = np.array(Image.open(input_image_p))
im_out = np.array(Image.open(out_png_p))
ssim, msssim, psnr = compare_imgs.compare(im_in,
im_out,
calc_ssim=True,
calc_msssim=True,
calc_psnr=True)
fout_str = _append_to_measures_f(fout, bpp, actual_bpp, ssim,
msssim, psnr)
print(fout_str, end='\r')
os.remove(output_image_j2_p)
print()