github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 01:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Converter: fixed degrade color power after 'lct'

Browse source
This commit is contained in:
iperov 2019-04-28 19:34:56 +04:00
parent ab714dfbfe
commit 7140ee8684
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
imagelib/reduce_colors.py
View file

@ -4,7 +4,7 @@ from PIL import Image
#n_colors = [0..256]
def reduce_colors (img_bgr, n_colors):
img_rgb = (cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB) * 255.0).astype(np.uint8)
img_rgb = (img_bgr[...,::-1] * 255.0).astype(np.uint8)
img_rgb_pil = Image.fromarray(img_rgb)
img_rgb_pil_p = img_rgb_pil.convert('P', palette=Image.ADAPTIVE, colors=n_colors)