Spaces:
Runtime error
Runtime error
File size: 681 Bytes
fadb92b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import os
import cv2
import numpy as np
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)
|