search

lobantseff / segmentation-mask-overlay

Plotting the segmentation masks has never been so exciting!
MIT License
8 stars 1 forks source link

Is handling of RGB Images Covered for np arrays? #1

Closed khanfarhan10 closed 1 year ago

khanfarhan10 commented 1 year ago

@lobantseff

I find AssertionError: Supported only grayscale numpy arrays.

Is handling of RGB Images Covered in the specific case of numpy arrays?

lobantseff commented 1 year ago

Hi @khanfarhan10

Yes. So far are only grayscale images supported, since I'm mostly use this library for medical imaging tasks. If you would like to add RGB, please feel free to propose a PR.

lobantseff commented 1 year ago

Hi @khanfarhan10,

I'm updating and simplifying the interface in move-to-cv2 branch. It also supports RGB numpy arrays now. If you're still interested, you can check it out! Performace also increased from 0.2 s with PIL-based in-paintings to 0.01 s with opencv addWeighted method.