TypeError: MaskAnnotator.__init__() got an unexpected keyword argument 'color_map'

[aimanmutasem]

Search before asking

• I have searched the Supervision issues and found no similar bug report.

Bug

Hello, thank you for the grate repo. I'm following the tutorial: https://blog.roboflow.com/how-to-use-segment-anything-model-sam/. The problem is when I want to draw masks on an image using provided detections with Segment-Anything (SAM).

The error is:
`
TypeError Traceback (most recent call last)
in <cell line: 1>()
----> 1 mask_annotator = sv.MaskAnnotator(color_map = "index")
2
3 detections = sv.Detections.from_sam(sam_result)
4
5 annotated_image = mask_annotator.annotate(scene=image_bgr.copy(), detections=detections)

TypeError: MaskAnnotator.init() got an unexpected keyword argument 'color_map'
`

Environment

No response

Minimal Reproducible Example

mask_annotator = sv.MaskAnnotator(color_map = "index")

detections = sv.Detections.from_sam(sam_result)

annotated_image = mask_annotator.annotate(scene=image_bgr.copy(), detections=detections)

sv.plot_images_grid(
images=[image_bgr, annotated_image],
grid_size=(1, 2),
titles=['source image', 'segmented image']
)

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[onuralpszr]

Hello, I moved issue to Q&A because it is not a bug and color_map become color_lookup and example use case

mask_annotator = sv.MaskAnnotator(color_lookup = sv.ColorLookup.INDEX) # INDEX # CLASS #TRACK

Enum is here : https://github.com/roboflow/supervision/blob/develop/supervision/annotators/utils.py#L11

[aimanmutasem]

Thank you @onuralpszr, mask_annotator = sv.MaskAnnotator(color_lookup = sv.ColorLookup.INDEX) # INDEX # CLASS #TRACK works with me..

[onuralpszr]

@aimanmutasem please mark as an answer :)