search

Joker316701882 / Salient-Object-Detection

This is tensorflow implementation for cvpr2017 paper "Deeply Supervised Salient Object Detection with Short Connections"
443 stars 139 forks source link

HOW to train #3

Open xiaojidan opened 6 years ago

xiaojidan commented 6 years ago

Hi @Joker316701882 I want to train my own data set, How I can do? Can you share the training codes?

Joker316701882 commented 6 years ago

@xiaojidan Sorry this is confidential so far. Modified a little according to paper "deeply supervised salient object detection with short connections"

Sucran commented 6 years ago

I have another little question in your training if your code is confidential. Did you really set the same learning rate for all layers and use the same data set from the paper "deeply supervised salient object detection with short connections"？ If so, I'm curious how to fix the overfitting problem in training this model？

Joker316701882 commented 6 years ago

@Sucran Learning rate is set 1e-5 for pixel-level loss all layers all time. And this model combines the encoder-decoder structure mentioned in "Deep Image matting". Train set is MSRA10K. I didn't encounter overfitting problem (random crop is the only augmentation trick applied here). Just left program run one night then got this model. It can be improved if apply more tricks mentioned in these years' segmentation paper.