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 ob

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-optimization-techniques-to-improve-separable-image-filters/ https://bartwronski.com/2020/03/15/using-jax-numpy-and-optim... HN Discussion: https://news.ycombinator.com/item?id=22590360&ref=hvper.com&utm_source=hvper.com&utm_medium=website https://news.ycombinator.com/item?id=22590360&ref=hvper.com&...