How can I train on my own dataset?
Engineering-Course opened this issue · comments
I want to train the model with my own dataset.
I'm confused about the input and output forms of the image data.
Could you help me?
Could you please share the training prototxt files and the solver configuration?
Thanks.
Please check the doc for training. The training script will generate the network and solver configurations.
The complete training procedure is described at https://github.com/fyu/dilation/blob/master/docs/training.md. Let me know there is any further questions about it.