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

How to train DAVIS datasets? #12

Closed Andy-zhujunwen closed 4 years ago

ptrvilya commented 4 years ago

If you want to train on any dataset you have to implement a dataset class similar to SBDDataset, and use it in the training script here.