File size: 682 Bytes
424188c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os
import numpy as np
import cv2
source = '../Structured3D/montefloor_data/test/'
dst = './viz_density'
for dirname in sorted(os.listdir(source)):
density_path = os.path.join(source, dirname, 'density.png')
density_img = cv2.imread(density_path)
density = 255 - density_img
out_path = os.path.join(dst, dirname + '.png')
out_alpha_path = os.path.join(dst, dirname + '_alpha.png')
alphas = np.zeros([density.shape[0], density.shape[1], 1], dtype=np.int32)
alphas[density_img.sum(axis=-1) > 0] = 255
density_alpha = np.concatenate([density, alphas], axis=-1)
cv2.imwrite(out_path, density)
cv2.imwrite(out_alpha_path, density_alpha) |