search

THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
799 stars 60 forks source link

About pretrain weights #47

Closed xiyouji6 closed 6 months ago

xiyouji6 commented 7 months ago

I used repvit_m1_5 as the backbone network and Retinanet head as the detection head to train on the target detection data set, but I did not find the pre-training weights in the code (checkpoint='pretrain/repvit_m1_5_distill_300e.pth'). I used the weights marked in red in the figure below as pre-training weights. but the weight values ​​do not match, so I would like to ask where to download the corresponding weights（pretrain/repvit_m1_5_distill_300e.pth）? ![Uploading 图片1.png…]()

jameslahm commented 7 months ago

Thanks for your interest. The image you uploaded is invisible. Would you mind updating the figure?

xiyouji6 commented 7 months ago

Sorry, the picture I uploaded cannot be displayed normally. The weight I used is Ckpt (M1.5) in RepViT-M1.5 under the Models module in this link.https://github.com/THU-MIG/RepViT/tree/main/detection