search

NVlabs / FreeSOLO

FreeSOLO for unsupervised instance segmentation, CVPR 2022
Other
312 stars 32 forks source link

Why is the number of categories for training coco 2 instead of 81? #4

Open APeiZou opened 2 years ago

APeiZou commented 2 years ago

Hello,@WXinlong Why is the number of categories for training coco 2 instead of 81?

lifuguan commented 2 years ago

In pretrain procedure, only foreground and background is unsed

APeiZou commented 2 years ago

Thanks a lot @lifuguan Why is the number of categories also 2 when training the self-supervise SOLO module, and the map calculated at this time has only two categories of maps?

lifuguan commented 2 years ago

Because both freemask and self-sup training are total the same, except for the dataset. The annotations of self-sup cames from the predictions of freemask training. It's general in self-sup training.

ztt0821 commented 2 years ago

Hi, I still feel confused. If the classes is 2 how to get the final category result?