CVMI-Lab / SimGCD

(ICCV 2023) Parametric Classification for Generalized Category Discovery: A Baseline Study
https://arxiv.org/abs/2211.11727
MIT License
85 stars 13 forks source link

Could you release the code for reimplementing UNO and RS? #9

Closed ycwang-libra closed 8 months ago

ycwang-libra commented 11 months ago

As same as your issue 'https://github.com/sgvaze/generalized-category-discovery/issues/20' in work 'GCD', could you please provide the code for implementing UNO and RS for further research?

xwen99 commented 9 months ago

Hi, we actually did not reimplement RS, and just took the results from GCD. Regarding UNO, the reproduction is quite easy, the major modification is to replace the pseudo-labelling part with sinkhorn (you may find the implementation of this function from UNO or SwAV's official codebase). You may also align other details according to your needs (eg, representation learning). Hope that is helpful.