#!/usr/bin/env python
import numpy as np
import matplotlib.pyplot as plt

import caffe
import caffe.proto.caffe_pb2 as cpb

from cogswell import keyboard

import recon.config as config
from recon import *

def main():
    '''
    Usage:
        vis.py <blob_name>
    '''
    import docopt, textwrap
    main_args = docopt.docopt(textwrap.dedent(main.__doc__))

    blob_name = main_args['<blob_name>']

    rec = Reconstructor('caffenet_1000')
    for i in range(96):
        rec.canonical_image(blob_name, i, 5, '/tmp/test_{}.jpg'.format(i))


if __name__ == '__main__':
    main()

@@ -244,16 +244,25 @@
   vis.py
+ %3Cnet_id%3E
  %3Cblob_n
@@ -266,16 +266,16 @@
 b_name%3E%0A
-
     '''%0A
@@ -364,16 +364,51 @@
 oc__))%0A%0A
+    net_id = main_args%5B'%3Cnet_id%3E'%5D%0A
     blob
@@ -469,23 +469,14 @@
 tor(
-'caffenet_1000'
+net_id
 )%0A  
@@ -496,10 +496,10 @@
 nge(
-96
+30
 ):%0A 

test script takes any net