Open ngenne opened 6 years ago
Please carefully read examples. You do not need to use Keras APIs.
from convnet_drawer import Model, Conv2D, MaxPooling2D, Flatten, Dense
from matplotlib_util import save_model_to_file
classifier = Model(input_shape=(1024, 1024, 1))
classifier.add(Conv2D(512, (256,256), strides=(4, 4), padding="same"))
classifier.add(MaxPooling2D(pool_size=(2, 2)))
classifier.add(Conv2D(512, (256,256), strides=(4, 4), padding="same"))
classifier.add(Conv2D(512, (256,256), strides=(4, 4), padding="same"))
classifier.add(MaxPooling2D(pool_size=(2, 2)))
classifier.add(Flatten())
classifier.add(Dense(4096))
classifier.add(Dense(4096))
classifier.add(Dense(1000))
save_model_to_file(classifier, "example.pdf")
The above code is enough. But conv256x256 is not expected (usually 3x3 - 11x11) thus the resulting image seems to be broken.
Hi there,
I'm getting an error everytime I want to generate the graph into the pdf file:
Below my code:
Do you have a trick for me?