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)
Пример #2
0
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 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()
Пример #4
0
 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)