xtudbxk/SEC-tensorflow

Trained Model, Please

terenceylchow124 opened this issue · 5 comments

Thanks for the tensorflow of SEC! I did the training, but the predicted image were all black images. I guess there might be something wrong of my dataset. Therefore, it would be helpful if you could provide the trained model also?

It seems I didn't keep the trained model now. And in my experience, you can adjust the lr smaller to avoid the problem. What's more, please remember to convert your dataset using the convert.py under "data" folder.

I did the procedure again for converting the dataset, and put it under "/data".
This time, I tried the evaluation code deeplab_aspp_test.py, but I though the create_network function went wrong since there were no "fc_1", "fc_2", "fc_3" and "fc_4". Even though I fixed this, the code still gave me error message:

image

the error happened from:

image

I guessed maybe something in dataset.py went wrong, but still couldn't find.

Thanks for the tensorflow of SEC! I did the training, but the predicted image were all black images. I guess there might be something wrong of my dataset. Therefore, it would be helpful if you could provide the trained model also?

I made it wrong, it should another repository, "DSRG-tensorflow". I trained the model again, and the result was very reliable which is comparable with the image predicted by the given pretrained model. But the "SEC-tensorflow" still faces some bugs in evaluation part which mentioned above.

Sorry for the late reply. The origin file "deeplab_aspp_test.py" is prepared for the deeplab-assp network while SEC uses the deeplab network. So you can modify the build function of evaluation code following [SEC.py](https://github.com/xtudbxk/SEC-tensorflow/blob/master/SEC.py}. The error you reported seems that the output of network has a error type, and it most probably related to the building network structure in build function. So I couldn't find the reason without the modifed code you did.

reopen if any question~