david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Colombo
2019-09-06 14:52:46 +04:00
parent faa14c4e59
commit 70dada42ea
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
imagelib/common.py
+4
View File
@@ -12,8 +12,12 @@ def normalize_channels(img, target_channels):
if c == 0 and target_channels > 0: if c == 0 and target_channels > 0:
img = img[...,np.newaxis] img = img[...,np.newaxis]
c = 1
if c == 1 and target_channels > 1: if c == 1 and target_channels > 1:
img = np.repeat (img, target_channels, -1) img = np.repeat (img, target_channels, -1)
c = target_channels
if c > target_channels: if c > target_channels:
img = img[...,0:target_channels] img = img[...,0:target_channels]
c = target_channels c = target_channels