How to train pytorch_deeplab_resnet by using my own data?

isht7 / pytorch-deeplab-resnet

DeepLab resnet v2 model in pytorch

MIT License

602 stars 118 forks source link

How to train pytorch_deeplab_resnet by using my own data? #3

Closed yghlc closed 7 years ago

yghlc commented 7 years ago

Hello, I have my own data, also can be converted to the format like VOC2012. But my data have only two classes such as the target and the background. How to set the number of class? Thanks

isht7 commented 7 years ago

@yghlc A new branch with custom data support has been added. During training and evaluation new flag "NoLabels" is added(more in README, and do python train.py -h for information about flag.). Please check if it serves your purpose. This branch would be merged with master after some more testing.

isht7 commented 7 years ago

Now, the master branch can also be used to train using your own data.