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

Can I run this model on my own data with only image and label? #24

Open Sysilia opened 3 months ago

Sysilia commented 3 months ago

Your work is great! I would like to switch to my own data to run your experiment, but I see that the input inside your code also requires graph files, my data is only image and road gt, how should I generate the graph myself?

htcr commented 3 months ago

Do you mean you only have GT masks for the roads? I think you have a few options, like: 1) Try to query some open-source map dataset like OpenStreetMap by location, see if you can get GT graphs. 2) You can also use heuristic processing to convert gt masks to gt graphs. Ideally also with some manual refinement.