rishizek / tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow
MIT License
286 stars 102 forks source link

Training a custom dataset without a pre-trained model #20

Open abderhasan opened 5 years ago

abderhasan commented 5 years ago

Hello,

Is there a way to train the model on a dataset from scratch? That is, without using a pre-trained model?

I believe my question is similar to a previous question, here.

Thanks a lot.

rishizek commented 5 years ago

Hi @abderhasan ,

Thank you for your interest in the repo. I think the easiest but dirty way to train the model from scratch is just commenting out these 2 lines.

I hope this helps.

abderhasan commented 5 years ago

Hello @rishizek,

Looks great! Thanks so much for your kind support.

zsfVishnu commented 5 years ago

@abderhasan , did it help? Are you able to train your custom data by commenting out those lines?