K-Mike / Automatic-salt-deposits-segmentation

MIT License
24 stars 11 forks source link

pretrainedmodels and pretrained weight #1

Open dfayzur opened 5 years ago

dfayzur commented 5 years ago

Hi,

Thanks for the code base. Do you have any plan to include the pretrainedmodels pythin file and pretrained weight on imagenet in the repo?

Diyago commented 5 years ago

you can get them https://github.com/qubvel/segmentation_models.pytorch and here https://github.com/qubvel/segmentation_models (keras)

liburning commented 4 years ago

hi, Can you write a demo for us to show how to train the dataset? Thank you!