yichen-purdue's picture
update the exmaples
2e7060e
raw
history blame contribute delete
No virus
322 Bytes
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)