Training it on custom dataset

SamsungLabs / fbrs_interactive_segmentation

[CVPR2020] f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation https://arxiv.org/abs/2001.10331

Mozilla Public License 2.0

583 stars 94 forks source link

Training it on custom dataset #6

Closed humzaahmad906 closed 4 years ago

humzaahmad906 commented 4 years ago

I want to train the given network on my own dataset. What steps do i need to follow?

ptrvilya commented 4 years ago

You have to define dataset class for your data (see SBDDataset for an example) and then create a model training script for your dataset (see r34_dh128 for an example).

humzaahmad906 commented 4 years ago

while training the network on custom dataset it gets stuck. Can you help me with it?