github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 02:37:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

+imagelib.normalize_channels

Browse source
This commit is contained in:
iperov 2019-03-30 21:52:36 +04:00
commit c1117852d8
2 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
imagelib/__init__.py
View file

@ -18,4 +18,6 @@ from .color_transfer import color_hist_match
from .color_transfer import reinhard_color_transfer
from .color_transfer import linear_color_transfer
from .DCSCN import DCSCN
from .DCSCN import DCSCN
from .common import normalize_channels

19
imagelib/common.py Normal file
View file

@ -0,0 +1,19 @@
def normalize_channels(img, target_channels):
img_shape_len = len(img.shape)
if img_shape_len == 2:
h, w = img.shape
c = 0
elif img_shape_len == 3:
h, w, c = img.shape
else:
raise ValueError("normalize: incorrect image dimensions.")
if c == 0 and target_channels > 0:
img = img[...,np.newaxis]
if c == 1 and target_channels > 1:
img = np.repeat (img, target_channels, -1)
if c > target_channels:
img = img[...,0:target_channels]
c = target_channels
return img