def __init__(self, fig, data): self.data = data self.fig = fig vi.visualize_encodings(data, grid=(3, 5), skip_every=5, fast=fast, fig=fig, interactive=True) plt.subplot(155).set_title(', '.join('hold on')) # fig.canvas.mpl_connect('button_press_event', self.on_click) fig.canvas.mpl_connect('pick_event', self.on_pick) try: # if True: ut.print_info('Checkpoint: %s' % FLAGS.load_from_checkpoint) self.model = dm.DoomModel() self.reconstructions = self.model.decode(data) except: ut.print_info("Model could not load from checkpoint %s" % str(sys.exc_info()), color=31) self.original_data, _ = inp.get_images(FLAGS.input_path) self.reconstructions = np.zeros(self.original_data.shape).astype( np.uint8) ut.print_info('INPUT: %s' % FLAGS.input_path.split('/')[-3]) self.original_data, _ = inp.get_images(FLAGS.input_path)
def visualize_latest_from_visualization_folder(folder='./visualizations/', file=None): if file is None: file = ut.get_latest_file(folder, filter=r'.*\d+\.txt$') ut.print_info('Encoding file: %s' % file.split('/')[-1]) data = np.loadtxt(file) # [0:360] fig = plt.figure() vi.visualize_encodings(data, fast=fast, fig=fig, interactive=True) fig.suptitle(file.split('/')[-1]) fig.tight_layout() plt.show()
def __init__(self, fig, data): self.data = data self.fig = fig vi.visualize_encodings(data, grid=(3, 5), skip_every=5, fast=fast, fig=fig, interactive=True) plt.subplot(155).set_title(', '.join('hold on')) # fig.canvas.mpl_connect('button_press_event', self.on_click) fig.canvas.mpl_connect('pick_event', self.on_pick) try: # if True: ut.print_info('Checkpoint: %s' % FLAGS.load_from_checkpoint) self.model = dm.DoomModel() self.reconstructions = self.model.decode(data) except: ut.print_info("Model could not load from checkpoint %s" % str(sys.exc_info()), color=31) self.original_data, _ = inp.get_images(FLAGS.input_path) self.reconstructions = np.zeros(self.original_data.shape).astype(np.uint8) ut.print_info('INPUT: %s' % FLAGS.input_path.split('/')[-3]) self.original_data, _ = inp.get_images(FLAGS.input_path)