search

yu4u / convnet-drawer

Python script for illustrating Convolutional Neural Networks (CNN) using Keras-like model definitions
MIT License
596 stars 98 forks source link

Support keras model #2

Closed wakame1367 closed 6 years ago

wakame1367 commented 6 years ago

Created a util function to use keras model class for input. Only Conv2D, MaxPooling2D, Flatten, Dense layers are supported.