htcr / sam_road

Segment Anything Model for large-scale, vectorized road network extraction from aerial imagery. CVPRW 2024
https://arxiv.org/pdf/2403.16051.pdf
MIT License
98 stars 13 forks source link

from detectron2.modeling import ViT #5

Closed wliuxing123 closed 1 month ago

wliuxing123 commented 3 months ago

May I ask where there is vit in the modeling? I downloaded detectron2,but without vit.

htcr commented 3 months ago

https://github.com/facebookresearch/detectron2/blob/main/detectron2/modeling/backbone/vit.py

notice this is not required for SAM-road, it's just an ablation study.

htcr commented 1 month ago

Closing, feel free to reopen if need further discussion.