[joshvm]

I meant about where this line comes from:

    mask = masks[0][0]  

Presumably 0 is the class ID? For someone new to ML or object detection, it might not be obvious why you take the first channel here.

Also recent related reading: https://bartwronski.com/2020/03/15/using-jax-numpy-and-optim...

HN Discussion: https://news.ycombinator.com/item?id=22590360&ref=hvper.com&...