keras-segnet
keras-segnet copied to clipboard
converted model building to use the keras2 api
Hi,
Keras 1 and Keras 2 don't really have compatible API.
Cleaned it up to use Keras2 API.
Let me know what you think. :)
Interesting Facts:
- keras 1 is fine with epsilon < 1e-5, but not keras2. The model generated with keras 2 is using epsilon 0.001.
- If you try to load the model.json with keras2, this will generate errors as described here: https://github.com/imlab-uiip/keras-segnet/issues/9. This fixes it. But it forces you to be on keras2
Hi @ytsheng, Looks good, thanks for your work. I will try to merge and adapt it by the end of the week.
Hi would you please say about the specification of the GPU in which the segnet model was trained? Also, I want to know about the batch size and the size of the input images.
Thanks
It did not seem to be merged so closing.