def save_fantasy(step, drec, cfg, pnum): plt.close('all') img_name = make_img_name("fantasy_%05d.png" % step) visualize_rows( pnum, drec, range(20), lambda x: make_img(x, cfg.px, cfg.py, cfg.maps_bottom, mbs, True), title='Fantasy/Reconstruction ' + str(step + 1), use_imshow=cfg.maps_bottom > 1, save=True, save_filename=img_name, normalize=True)
def save_fantasy(step,drec,cfg,pnum): plt.close('all') img_name = make_img_name("fantasy_%05d.png"%step) visualize_rows(pnum,drec,range(20), lambda x:make_img(x,cfg.px,cfg.py,cfg.maps_bottom,mbs,True), title='Fantasy/Reconstruction '+str(step+1), use_imshow=cfg.maps_bottom>1, save=True, save_filename=img_name, normalize=True)
#### calculate maps_bottom into py. yeah it's a dirty hack, i know px = cfg.px py = cfg.py if "projection_results" in rbmstack.__dict__: for layernum in rbmstack.projection_results.keys(): filters = rbmstack.projection_results[layernum].T print "Saving projections from layer %d (%d x %d)" % ( layernum, filters.shape[0], filters.shape[1]) img_name = make_img_name("filter_layer%d.png" % (layernum)) visualize_rows(PLT_NUM, filters, range(20), lambda x: make_img(x, px, py, cfg.maps_bottom, True), title='Projection W Layer %d' % layernum, use_imshow=cfg.maps_bottom > 1, save=True, save_filename=img_name, normalize=False) PLT_NUM += 1 sys.exit(0) def save_fantasy(step, drec, cfg, pnum): plt.close('all') img_name = make_img_name("fantasy_%05d.png" % step) visualize_rows( pnum, drec, range(20),
if cfg.headless: cp.exitCUDA() sys.exit(0) PLT_NUM=1 import matplotlib.pyplot as plt #### calculate maps_bottom into py. yeah it's a dirty hack, i know px=cfg.px py=cfg.py if "projection_results" in rbmstack.__dict__: for layernum in rbmstack.projection_results.keys(): filters = rbmstack.projection_results[layernum].T print "Saving projections from layer %d (%d x %d)" % (layernum,filters.shape[0],filters.shape[1]) img_name = make_img_name("filter_layer%d.png"%(layernum)) visualize_rows(PLT_NUM,filters,range(20), lambda x:make_img(x,px,py,cfg.maps_bottom,True), title='Projection W Layer %d'%layernum, use_imshow=cfg.maps_bottom>1, save=True, save_filename=img_name, normalize=False) PLT_NUM+=1 sys.exit(0) def save_fantasy(step,drec,cfg,pnum): plt.close('all') img_name = make_img_name("fantasy_%05d.png"%step) visualize_rows(pnum,drec,range(20), lambda x:make_img(x,cfg.px,cfg.py,cfg.maps_bottom,mbs,True), title='Fantasy/Reconstruction '+str(step+1), use_imshow=cfg.maps_bottom>1, save=True, save_filename=img_name, normalize=True) cb = lambda step, drec: save_fantasy(step,drec,cfg,PLT_NUM) rbmstack.prepare_dbg(mbp,30,cfg.eval_steps,cfg.eval_start,cb) PLT_NUM += 1 fan=np.array(rbmstack.dbg_datout) if "W" in rbmstack.__dict__ : W_old=rbmstack.W.copy()