diff options
Diffstat (limited to 'megapixels/app/utils/im_utils.py')
| -rw-r--r-- | megapixels/app/utils/im_utils.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/megapixels/app/utils/im_utils.py b/megapixels/app/utils/im_utils.py index d5e92aa3..e882c67f 100644 --- a/megapixels/app/utils/im_utils.py +++ b/megapixels/app/utils/im_utils.py @@ -21,6 +21,12 @@ import datetime +def num_channels(im): + '''Returns number of channels in numpy.ndarray image''' + if len(im.shape) > 2: + return im.shape[2] + else: + return 1 def is_grayscale(im, threshold=5): """Returns True if image is grayscale |