consistent pixel value for an object

[mikeyEcology]

Hi, Thank you for creating this tutorial. I'm wondering if there is a way to modify the script so that each pixel value for an object is the same. For example, if I have an object that is category id 4, the current script produces a mask that has values of 0, 1, 2, 3, 4. But I want all pixels to be either 0 (background) or 4 (object).

I tried modifying the script to do this, but I'm pretty confused when it comes to using the coco library.

Thank you