search

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
150 stars 18 forks source link

from detectron2.modeling import ViT #5

Closed wliuxing123 closed 6 months ago

wliuxing123 commented 7 months ago

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

htcr commented 7 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 6 months ago

Closing, feel free to reopen if need further discussion.