diff options
| author | adamhrv <adam@ahprojects.com> | 2019-01-03 12:51:31 +0100 |
|---|---|---|
| committer | adamhrv <adam@ahprojects.com> | 2019-01-03 12:51:31 +0100 |
| commit | f215db6e84071077082d14f8366ae1cf1aea500f (patch) | |
| tree | 33e4573eb618f21685809cf567fdf196ff673a91 /megapixels/app/utils/im_utils.py | |
| parent | 5340bee951c18910fd764241945f1f136b5a22b4 (diff) | |
fix roi index, clean up pose, roi, records, vector
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 |