1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
import os
import time
import subprocess
import argparse
start_time = time.time()
parser = argparse.ArgumentParser()
parser.add_argument('--folder', required=True)
parser.add_argument('--out_fn', required=True)
parser.add_argument('--dataset', required=True)
opt = parser.parse_args()
os.makedirs('./uprez', exist_ok=True)
dataset = opt.dataset
print(">>> {}".format(dataset))
in_dir = os.path.abspath(opt.folder)
vsr_duf_inputs_dir = os.path.abspath('../vsr-duf/inputs')
vsr_duf_results_dir = os.path.abspath('../vsr-duf/results/{}'.format(dataset))
out_dir = os.path.abspath('./uprez')
subprocess.call([
'/home/lens/miniconda3/envs/tf/bin/python',
'../vsr-duf/crop_frames.py',
'--in_dir', in_dir,
'--out_dir', vsr_duf_inputs_dir,
'--ratio', '6.0'
])
subprocess.call([
'/home/lens/miniconda3/envs/tf/bin/python',
'../vsr-duf/test.py',
'--in_dir', vsr_duf_inputs_dir,
'--out_dir', vsr_duf_results_dir,
'--network_dir', os.path.abspath('../vsr-duf'),
])
subprocess.call([
'/home/lens/miniconda3/envs/tf/bin/python',
'../vsr-duf/merge_frames.py',
'--in_dir', vsr_duf_results_dir,
'--render_dir', out_dir,
'--out_fn', opt.out_fn,
'--mov',
])
finish_time = time.time()
print("took %02f s.".format(finish_time - start_time))
|
