def test_load_png_file(self):
'it should return an Image Type if file exists and is an Image file'
# case 1 (it should return PngImageFile):
img = load_img(os.path.join(FOLDER, RIGHT_FILENAME))
self.assertEqual(type(img).__name__, 'PngImageFile')
# case 2 (it should return NoneType):
img = load_img(os.path.join(FOLDER, WRONG_FILENAME))
self.assertEqual(type(img).__name__, 'NoneType')
def test_img_from_dwt_coeff(self):
'TODO'
img = load_img(os.path.join(FOLDER, FILENAME))
coeff = extract_rgb_coeff(img)
img_new = img_from_dwt_coeff(coeff)
img_new.save('/home/jiewend/download/a_4component.png')
return
def test_extract_rgb_coeff(self):
'it should return a Tuple of coefficient per each channel (RGB)'
img = load_img(os.path.join(FOLDER, FILENAME))
(c_r, c_g, c_b) = extract_rgb_coeff(img)
self.assertTrue(c_r[0] is not None)
self.assertTrue(c_r[1] is not None)
self.assertTrue(c_g[0] is not None)
self.assertTrue(c_g[1] is not None)
self.assertTrue(c_b[0] is not None)
self.assertTrue(c_b[1] is not None)
def test_creat_lost_dwt_coef(self):
img = load_img(os.path.join(FOLDER, FILENAME))
coeff = extract_rgb_coeff(img)
coff_lost = creat_lost_LH_HL_HH_dwt_coef(coeff, 1)
img_new = recontract_img_from_dwt_coef(coff_lost)
img_new.save('/home/jiewend/download/a_re_lost_LH_HL_HH.png')
coff_lost = creat_lost_HL_HH_dwt_coef(coeff, 1)
img_new = recontract_img_from_dwt_coef(coff_lost)
img_new.save('/home/jiewend/download/a_re_lost_HL_HH.png')
coff_lost = creat_lost_HH_dwt_coef(coeff, 1)
img_new = recontract_img_from_dwt_coef(coff_lost)
img_new.save('/home/jiewend/download/a_re_lost_HH.png')
def _get_image(self, img_path):
raw_img = load_img(img_path,
to_rgb=True,
color_img=self.data_conf['color_image'])
height, width = raw_img.shape[:2]
if height % 2**self.num_scales != 0 or width % 2**self.num_scales != 0 or self.data_conf[
'use_crop']:
crop_size = int(self.data_conf['crop_size']
) if self.data_conf['use_crop'] else min(
height, width)
crop_size = crop_size // 2**self.num_scales * 2**self.num_scales
raw_img = central_crop(raw_img, crop_size)
img_name = os.path.basename(img_path)
save_img(raw_img, os.path.join(self.originals_dir, img_name))
return raw_img
def test_recontract_img_from_dwt_coef(self):
img = load_img(os.path.join(FOLDER, FILENAME))
coeff = extract_rgb_coeff(img)
coeff_copy = coeff
img_new = recontract_img_from_dwt_coef(coeff)
def test_yuv_to_rgb(self):
'it should return an Image Type after conversion'
img = load_img(os.path.join(FOLDER, FILENAME))
img_yuv = yuv_to_rgb(img)
self.assertEqual(type(img_yuv).__name__, 'Image')
def test_img_from_dwt_coeff(self):
'TODO'
img = load_img(os.path.join(FOLDER, FILENAME))
coeff = extract_rgb_coeff(img)
return
def test_img_loader():
img_list = os.listdir(config.dataset_dir)
img = utils.load_img(img_list[0], config.resolution)
def run():
img = util.load_img(os.path.join(FOLDER, RIGHT_FILENAME))