COCO dataset!!! Can not load coco dataset for mask image

[chenhao-zju]

When I read the mask image, I found that the category corresponding to the mask obtained by reading does not correspond to the category of the coco dataset. How can I solve this problem?
Thanks

[zhiheLu]

Did you generate the masks by this code: https://github.com/mboudiaf/RePRI-for-Few-Shot-Segmentation/blob/master/data/coco/create_masks.py?