Function main

dataset_loaders/seven_scenes.py:357–394

visualizes the dataset

()

def main():
  """
  visualizes the dataset
  """
  # from common.vis_utils import show_batch, show_stereo_batch
  from torchvision.utils import make_grid
  import torchvision.transforms as transforms
  seq = 'heads'
  mode = 1
  num_workers = 6
  transform = transforms.Compose([
    transforms.Scale(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
  ])
  target_transform = transforms.Lambda(lambda x: torch.Tensor(x))
  dset = SevenScenes(seq, '../data/deepslam_data/7Scenes', True, transform, target_transform=target_transform, mode=mode)
  print('Loaded 7Scenes sequence {:s}, length = {:d}'.format(seq, len(dset)))
  pdb.set_trace()

  data_loader = data.DataLoader(dset, batch_size=4, shuffle=True, num_workers=num_workers)

  batch_count = 0
  N = 2
  for batch in data_loader:
    print('Minibatch {:d}'.format(batch_count))
    pdb.set_trace()
    # if mode < 2:
    #   show_batch(make_grid(batch[0], nrow=1, padding=25, normalize=True))
    # elif mode == 2:
    #   lb = make_grid(batch[0][0], nrow=1, padding=25, normalize=True)
    #   rb = make_grid(batch[0][1], nrow=1, padding=25, normalize=True)
    #   show_stereo_batch(lb, rb)

    batch_count += 1
    if batch_count >= N:
      break

if __name__ == '__main__':
  main()