text: import argparse
  from pathlib import Path
  
  import torch
  from run import run
  
  if __name__ == "__main__":
  
      parser = argparse.ArgumentParser()
      parser.add_argument('--root', type=str, default='/home/daichi/Desktop/2dao/datasets/maya_v9_384')
      args = parser.parse_args()
  
      # set torch options
      torch.backends.cudnn.enabled = True
      torch.backends.cudnn.benchmark = True
  
      root = Path(args.root)
      assert root.exists()
  
      for subset in ['train', 'val', 'test']:
      # for subset in ['test']:
          run(input_path=str(root / subset / 'rgb'), output_path=str(root / subset / 'midas'), model_path='model.pt')
  