test_bgs = [f for f in os.listdir(bg_test) if
os.path.isfile(os.path.join(bg_test, f)) and f.endswith('.png')]
sample_bgs = random.sample(test_bgs, 10)
total_loss = 0.0
for i in range(len(samples)):
filename = samples[i]
image_name = filename.split('.')[0]
print('\nStart processing image: {}'.format(filename))
bgr_img = cv.imread(os.path.join(out_test_path, filename))
bg_h, bg_w = bgr_img.shape[:2]
print('bg_h, bg_w: ' + str((bg_h, bg_w)))
a = get_alpha_test(image_name)
a_h, a_w = a.shape[:2]
print('a_h, a_w: ' + str((a_h, a_w)))
alpha = np.zeros((bg_h, bg_w), np.float32)
alpha[0:a_h, 0:a_w] = a
trimap = generate_trimap(alpha)
different_sizes = [(320, 320), (320, 320), (320, 320), (480, 480), (640, 640)]
crop_size = random.choice(different_sizes)
x, y = random_choice(trimap, crop_size)
print('x, y: ' + str((x, y)))
bgr_img = safe_crop(bgr_img, x, y, crop_size)
alpha = safe_crop(alpha, x, y, crop_size)
trimap = safe_crop(trimap, x, y, crop_size)
cv.imwrite('images/{}_image.png'.format(i), np.array(bgr_img).astype(np.uint8))
out_test_path = 'merged_test/'
test_images = [f for f in os.listdir(out_test_path) if
os.path.isfile(os.path.join(out_test_path, f)) and f.endswith('.png')]
samples = random.sample(test_images, 10)
total_loss = 0.0
for i in range(len(samples)):
filename = samples[i]
image_name = filename.split('.')[0]
print('Start processing image: {}'.format(filename))
x_test = np.empty((1, img_rows, img_cols, 4), dtype=np.float32)
bgr_img = cv.imread(os.path.join(out_test_path, filename))
bg_h, bg_w = bgr_img.shape[:2]
print(bg_h, bg_w)
a = get_alpha_test(image_name)
a_h, a_w = a.shape[:2]
print(a_h, a_w)
alpha = np.zeros((bg_h, bg_w), np.float32)
alpha[0:a_h, 0:a_w] = a
trimap = generate_trimap(alpha)
different_sizes = [(320, 320), (320, 320), (320, 320), (480, 480), (640, 640)]
crop_size = random.choice(different_sizes)
x, y = random_choice(trimap, crop_size)
print(x, y)
bgr_img = safe_crop(bgr_img, x, y, crop_size)
alpha = safe_crop(alpha, x, y, crop_size)
trimap = safe_crop(trimap, x, y, crop_size)
cv.imwrite('images/{}_image.png'.format(i), np.array(bgr_img).astype(np.uint8))
cv.imwrite('images/{}_trimap.png'.format(i), np.array(trimap).astype(np.uint8))
cv.imwrite('images/{}_alpha.png'.format(i), np.array(alpha).astype(np.uint8))