added graph_reader.py

import tensorflow as tf

from argparse import ArgumentParser

def main():

    parser = ArgumentParser(

        description=''

    )

    parser.add_argument(

        '-g', '--graph', required=True, help='path to graph'

        )

    args = parser.parse_args()

    filename = args.graph

    gf = tf.GraphDef()

    gf.ParseFromString(open(filename, 'rb').read())

    with open(filename + '_layers.txt', 'w') as f:

        with open(filename + '_layers_for_lucid.txt', 'w') as w:

            for n in gf.node:

                if 'Concat' in n.op:

                    w.write(n.name + ' => ' + n.op + '\n')

                    f.write(n.name + ' => ' + n.op + '\n')

                f.write(n.name + ' => ' + n.op + '\n')

if __name__ == '__main__':

    main()