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
730 stars 55 forks source link

about semantic segmentation #10

Closed zhangh0920 closed 1 year ago

zhangh0920 commented 1 year ago

I wonder whether the model uses the re-parameter technique when processing semantic segmentation in the testing stage. Thanks a lot.

jameslahm commented 1 year ago

Thanks. In the testing stage, the re-parameter structure will be fused like https://github.com/THU-MIG/RepViT/blob/8ef1a71e299fc02ea5eaf201039f310313a66b96/segmentation/repvit.py#L105-L123.

zhangh0920 commented 1 year ago

Thanks for your apply. But I only see the re-parameter structure used in the classification task but I don't figure out in the segmentation task whether re-parameter structure is used.

jameslahm commented 1 year ago

In the segmentation task, re-parameter structure is also used like https://github.com/THU-MIG/RepViT/blob/8ef1a71e299fc02ea5eaf201039f310313a66b96/segmentation/repvit.py#L94-L123

zhangh0920 commented 1 year ago

I got it! Thanks a lot.