HR CustomDataset Finetuning

[plemeri]

HR CustomDataset Finetuning

[plemeri]

Hi @ZeVicTech, first of all, sorry for the late reply. I cannot guarantee that I can give you the right answer since I cannot access to your custom dataset and test myself, but please make sure that your dataset annotation strategy is identical to the salient object detection task.

For example, when your target is to generate a segmentation mask of a human hair only, then the rest of the public training datasets are not helpful for the finetuning. Try using your custom dataset only with lower learning rate, and start from our pre-trained model. You can refer to #40 for finetuning.

For the dynamic resize, we follow CascadePSP. It helps to produce better boundary accuracy by maintaining the aspect ratio of the image while reducing the size of the image to the plausible size for the GPU memory.

[ZeVicTech]

I'll give it a try as you advise, thank you