keras-segnet icon indicating copy to clipboard operation
keras-segnet copied to clipboard

converted model building to use the keras2 api

Open hackgoofer opened this issue 8 years ago • 2 comments

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

hackgoofer avatar Jun 11 '17 01:06 hackgoofer

Hi @ytsheng, Looks good, thanks for your work. I will try to merge and adapt it by the end of the week.

eclique avatar Jun 12 '17 13:06 eclique

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

Fahim-F avatar Jul 21 '17 00:07 Fahim-F

It did not seem to be merged so closing.

hackgoofer avatar Mar 22 '23 19:03 hackgoofer