TACJu / TransFG

This is the official PyTorch implementation of the paper "TransFG: A Transformer Architecture for Fine-grained Recognition" (Ju He, Jie-Neng Chen, Shuai Liu, Adam Kortylewski, Cheng Yang, Yutong Bai, Changhu Wang, Alan Yuille).
MIT License
382 stars 88 forks source link

About PSM module #37

Open SylU0 opened 1 month ago

SylU0 commented 1 month ago

I attempted to verify the functionality of the PSM module and noticed that incorporating only the PSM module into ViT didn't seem to enhance performance. Could you please let me know if there is a transfg model weights trained on CUB?