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

how to use my dataset? #2

Closed kuaiqushangzixiba closed 1 month ago

kuaiqushangzixiba commented 3 months ago

how to use my dataset?

htcr commented 3 months ago

Hi kuaiqushangzixiba:

You'll need to prepare the rgb images and GT graphs.

You can refer to the cityscale dataset of how to organize them.

You can refer to line 75-86 in cityscale/generate_labels.py for the format of GT graphs.

Specifically, as of now, this project assumes the following format of ground-truth graph: