colorize

omnipose.plot.colorize(im, colors=None, color_weights=None, offset=0, channel_axis=- 1)