search

hustvl / TopFormer

TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022
Other
373 stars 42 forks source link

is there any ImageNet pre-training config of Topformer? #17

Closed wmkai closed 1 year ago

wmkai commented 2 years ago

I want to change some modules when using Topformer to do classification on ImageNet. Could you please provide the ImageNet Pre-training config? Thanks!

speedinghzl commented 2 years ago

Please look into the close issue. You could obtain the classification model by removing SIM in the Seg model and adding a Linear layer on the top of the Transformer block (SASE) as the classification head.

wmkai commented 2 years ago

If it is convenient for you, can I have your wechat and share the code of imagenet pre-training?