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)
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')
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__ :
