search

SysCV / transfiner

Mask Transfiner for High-Quality Instance Segmentation, CVPR 2022
https://www.vis.xyz/pub/transfiner
Apache License 2.0
528 stars 60 forks source link

what is the pred_class index refer to? #27

Closed anl13 closed 2 years ago

anl13 commented 2 years ago

Hello, I wonder what is the output pred_class refer to? Does it follow COCO classes names just like https://gist.github.com/AruniRC/7b3dadd004da04c80198557db5da4bda?
When I run on the demo images, I found that the class index for people is 0, the class index for mouse is 15, is that correct?

lkeab commented 2 years ago

Hi, you can refer to the category orders here.

anl13 commented 2 years ago

interesting. It recognize mouse as 'cat', haha.