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

Number of epochs #10

Closed kurdt23 closed 6 months ago

kurdt23 commented 6 months ago

Hello, I would like to thank you for your excellent work. I read your article and ran the model at 10 epochs, but my metric is a little different from the one presented in the article. I could not find the exact number of epochs in the article. Please can you tell me what is the number of epochs you are using in the training in the article?

htcr commented 6 months ago

The released config is what we used for the paper. I'd expect small fluctuations if you re-train the model, should not change the overall conclusion.

htcr commented 6 months ago

I'll release our ckpt soon, you may compare your results with it too.

htcr commented 6 months ago

Closing, feel free to re-open if further discussion is needed.