fix

2019-09-06 14:52:46 +04:00
parent faa14c4e59
commit 70dada42ea
1 changed files with 4 additions and 0 deletions
--- a/imagelib/common.py
+++ b/imagelib/common.py
@@ -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