Closed summerrr closed 6 years ago
The infer function is pretty basic, it was written for a binary class situation, you might have to modify it depending on the values your prediction contains. The FC-DenseNet is also difficult to optimize for other datasets, see problems.
I have used this code to successfully train on a binary class situation, key things to keep in mind are that the model is sensitive to class imbalance, so if you have imbalanced classes (background dominates, then the model will tend to predict background), try the following:
I am not aware of other settings of the problem you are trying to solve, so this is about as much help as I can give.
after training,model will save three ckpt files, .index .meta and .data, how can I run infer?I run the
python main.py --ckpt=models/model.ckpt-4
but the pictures in output are all black. And there is no errors and warnings.