def test_yuv_rgb(self): img = load_img(os.path.join(FOLDER, FILENAME)) img = jpeg2000.dc_offset(img, 8) img_yuv = jpeg2000.rgb_yuv(img) print(img_yuv.min(), img_yuv.max()) img_rgb = jpeg2000.yuv_rgb(img_yuv) show_img([img_yuv, img_rgb], ["yuv", "rgb"], 2, 1)
def test_reconstruct_help(self): img = load_img(os.path.join(FOLDER, FILENAME)) coeffs_dwt = jpeg2000.extract_rgb_coeff(img) img = jpeg2000.reconstruct_help(coeff=coeffs_dwt.coeff_r, color_value=160) plt.imshow(img) plt.show()
def test_dc_offset(self): img = load_img(os.path.join(FOLDER, FILENAME)) B = get_bits(img) # img_dc = jpeg2000.dc_offset(img, B) img_dc = jpeg2000.dc_offset(img, B) plt.imshow(img_dc) plt.show()
def test_img_from_dwt_coeff(self): img = load_img(os.path.join(FOLDER, FILENAME)) coeffs_dwt = jpeg2000.extract_rgb_coeff(img) re_img = jpeg2000.img_from_dwt_coeff(coeffs_dwt) print(re_img.max()) print(re_img.shape) plt.imshow(re_img) plt.show()
def test_forward(self): img = load_img(os.path.join(FOLDER, FILENAME)) quatization_yuv = jpeg2000.forward(img)
def test_quantization(self): img = load_img(os.path.join(FOLDER, FILENAME)) img_yuv = jpeg2000.rgb_yuv(img) coeffs_yuv = jpeg2000.extract_yuv_coeff(img_yuv) quatization_yuv = jpeg2000.quatization(coeffs_yuv, 30) assert quatization_yuv is not None
def test_extract_yuv_coeff(self): img = load_img(os.path.join(FOLDER, FILENAME)) img = jpeg2000.dc_offset(img, 8) img_yuv = jpeg2000.rgb_yuv(img) coeffs_yuv = jpeg2000.extract_yuv_coeff(img_yuv) assert coeffs_yuv is not None
def test_rgb_yuv(self): img = load_img(os.path.join(FOLDER, FILENAME)) img_yuv = jpeg2000.rgb_yuv(img) print(img_yuv.min(), img_yuv.max())