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

fix

Browse source
This commit is contained in:
Colombo 2019-09-06 14:52:46 +04:00
commit 70dada42ea
1 changed files with 4 additions and 0 deletions

4
imagelib/common.py
View file

@ -12,8 +12,12 @@ def normalize_channels(img, target_channels):
if c == 0 and target_channels > 0:
img = img[...,np.newaxis]
c = 1
if c == 1 and target_channels > 1:
img = np.repeat (img, target_channels, -1)
c = target_channels
if c > target_channels:
img = img[...,0:target_channels]
c = target_channels