Training on Custom Data

[DecentMakeover]

Hi , Thanks for sharing your work, Could you please post an some guidelines/steps to train on custom data?

Thanks

[thekeviv]

Yes, I am also struggling to make this work on my own dataset

[AndrejHafner]

Did you have any success training on your own data?

[DecentMakeover]

ended up using this, https://github.com/CSAILVision/semantic-segmentation-pytorch. This has a HRNet Backbone support.

[InternetMaster1]

@sunke123

How to do training from scratch on a custom dataset (with base model without any pretraining on CityScapes or any other dataset) with this library for HRNETV2-Small-W18?

I was hoping I don't have to use a separate library to achieve this.

Thanks!

[ghost]

@InternetMaster1

If you are using this file, then it doesn't need any pretrained network. Every time you run the code, it runs from scratch.