"""
Describe a keras model (load it and create graph and description output)

keras_describe.py  model_file

"""


from __future__ import print_function
from keras.preprocessing.image import ImageDataGenerator
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation, Flatten
from keras.layers import Convolution2D, MaxPooling2D
from keras.optimizers import SGD, Adagrad, Adadelta, RMSprop
from keras.utils import np_utils
try:
    from keras.utils.visualize_util import plot
except ImportError:
    from keras.utils.vis_utils import plot_model as plot
from keras.models import load_model
from keras.models import model_from_json
from keras.callbacks import EarlyStopping, TensorBoard
import sys, random, pickle, numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import AxesGrid
from datetime import date


filename = sys.argv[1]

model = None
if filename[-3:3] in ['hd5', 'hdf', 'df5']:
	model = load_model(filename)
else:
	with open(filename, 'rU') as json_file:
		model = model_from_json(json_file.read())

print('Network Layout:')
model.summary()

plot_name = '.'.join(filename.split('.')[0:-1]) + ".png"

plot(model, to_file=plot_name, show_shapes=True)