diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 8 |
1 files changed, 6 insertions, 2 deletions
@@ -18,7 +18,7 @@ parser.add_argument('--L', metavar='L', type=int, default=28, help='Network dept parser.add_argument('--T', metavar='T', default='L', help='Input type: L(Low-resolution) or G(Ground-truth)') parser.add_argument('--dataset', default=None, help='Name of dataset') parser.add_argument('--in_dir', metavar='in_dir', default=None, help='Directory to process') -parser.add_argument('--out_dir', metavar='out_dir', default='./results', help='Directory to output to') +parser.add_argument('--out_dir', metavar='out_dir', default='./results/test', help='Directory to output to') parser.add_argument('--network_dir', default='.', help='Path to networks') parser.add_argument('--mov_from_dirs', action='store_true') args = parser.parse_args() @@ -47,13 +47,17 @@ def process_dir(dir): dataset = args.dataset = dir_partz[-2] part = dir_partz[-1] tag = '_'.join([dataset, str(args.L) + 'L', part]) - out_path = os.path.join(args.out_dir, dataset, part) + out_path = os.path.join(args.out_dir, part) render_path = os.path.join(args.out_dir, 'renders') + print('process dir {}, out path: {}'.format(dir, out_path)) os.makedirs(out_path, exist_ok=True) if args.mov_from_dirs: os.makedirs(render_path, exist_ok=True) dir_frames = sorted(glob.glob(os.path.join(dir, '*.png'))) + if not len(dir_frames): + print("{}: no frames!".format(dir)) + return # print(dir_frames) frames = [] |