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

Method of how to generate the GT graph #18

Closed Ikaros0123 closed 5 months ago

Ikaros0123 commented 5 months ago

Thanks for your great work!

Do I have to generate the GT graph if I just want to run inferencer.py using the given trained checkpoint?

I notice line 259, 263, 283 and 284 in inferencer.py, it seems like those code need GT graph of the rgb picture as input.

And if so, can you kindly share me the code/method to generate the GT graph? I would greatly appreciate it.

wfantastic commented 1 month ago

Hello, I would like to ask you this question. Have you found a way to generate GT graph? My own dataset only has the original road remote sensing image and labels. I hope you can reply me when you have time. Have a good life!