zorzi-s / PolyWorldPretrainedNetwork

PolyWorld: Polygonal Building Extraction with Graph Neural Networks in Satellite Images
Other
175 stars 28 forks source link

Train on custom datasets #15

Closed hehongjie closed 1 year ago

hehongjie commented 2 years ago

Thanks for your excellent work and share it here. I have tested it and it worked as expected. I am wondering if you can also share how to train on custom dataset here. Many thanks

AlbertoBlackPrint commented 1 year ago

Were you ever able to do it?

henokyen commented 1 year ago

I also want to retrain Polyworld with my own dataset. Can you please provide a train.py script that implements your end-to-end training approach?

xzw2 commented 1 year ago

I also want the train.py. Many thanks

lsjBatter commented 1 year ago

Thank you very much for sharing your work. Can you please provide a train.py script?

zorzi-s commented 1 year ago

@hehongjie , @AlbertoBlackPrint , @henokyen , @xzw2 , @lsjBatter

Thank you for the interest on our work! As I already mentioned in some closed Issues, making the training code public is not up to me unfortunately. The code is owned by blackshark.ai and they don’t want to release it fully.

hehongjie commented 1 year ago

Were you ever able to do it?

No. I can't.

hehongjie commented 1 year ago

@hehongjie , @AlbertoBlackPrint , @henokyen , @xzw2 , @lsjBatter

Thank you for the interest on our work! As I already mentioned in some closed Issues, making the training code public is not up to me unfortunately. The code is owned by blackshark.ai and they don’t want to release it fully.

Thanks for your reply. Looking forward to your new work.