search

maikherbig / AIDeveloper

GUI-based software for training, evaluating and applying deep neural nets for image classification
BSD 2-Clause "Simplified" License
109 stars 20 forks source link

Implementing last "very deep" models #9

Closed alfrem closed 4 years ago

alfrem commented 4 years ago

Just as an idea. Would be interesting to have AlexNet, ResNet or SqueezeNet.

Resnet can be tricky with parallelism and residual module but provide excelent results.

Thank you.

alfrem commented 4 years ago

I suppose this can be done in model_zoo.py, right?

maikherbig commented 4 years ago

yes :) just follow the example of the vgg19 or so. Let me know if it worked!

maikherbig commented 4 years ago

ResNet is now implemented: c3fc8b6 (for this to work you need keras applications >=1.0.7; this is installed by default when you install keras v >=2.2.4) The rest of AIDeveloper will now be adjusted to work with keras v. 2.2.4

maikherbig commented 4 years ago

SqueezeNet could be taken from here: https://github.com/rcmalli/keras-squeezenet

maikherbig commented 4 years ago

SqueezeNet is now added to the model_zoo: 3662af5

maikherbig commented 4 years ago

ResNet v2 is not working: Error: module 'keras.applications' has no attribute 'resnet'

maikherbig commented 4 years ago

Solved via 80696fc