Spaces:
Runtime error
Runtime error
import matplotlib.pyplot as plt | |
import numpy as np | |
rgb_file = 'fg-1-rgb.png' | |
alpha_file = 'fg-1-alpha.png' | |
output_file = 'fg-1-rgba.png' | |
rgb = plt.imread(rgb_file) | |
alpha = plt.imread(alpha_file) | |
print(rgb.shape, alpha.shape) | |
rgba = np.concatenate([rgb[..., :3], alpha[..., 0:1]], axis=2) | |
plt.imsave(output_file, rgba) | |