-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcreate_maps.py
58 lines (50 loc) · 2.13 KB
/
create_maps.py
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
54
55
56
57
58
import os
import glob
import argparse
import numpy as np
from scipy import misc
dir_name = 'maps-comparison'
methods = ['bigan', 'cogan', 'feature_loss', 'l1', 'cycle', 'pix2pix']
subdirs = ['aerial2map', 'map2aerial']
for filename in glob.glob('images/%s/%s/gt/*.jpg' % (dir_name,subdirs[0])):
nameonly = os.path.basename(filename)
direction = 0
subdir = subdirs[direction]
theotherdir = subdirs[1-direction]
print(("|  " +
"|  " +
"|  " +
"|  " +
"|  " +
"|  " +
"|  " +
"|  |") %
(dir_name,theotherdir,'gt',nameonly,
dir_name,subdir,methods[0],nameonly,
dir_name,subdir,methods[1],nameonly,
dir_name,subdir,methods[2],nameonly,
dir_name,subdir,methods[3],nameonly,
dir_name,subdir,methods[4],nameonly,
dir_name,subdir,methods[5],nameonly,
dir_name,subdir,'gt',nameonly,
))
direction = 1
subdir = subdirs[direction]
theotherdir = subdirs[1-direction]
print(("|  " +
"|  " +
"|  " +
"|  " +
"|  " +
"|  " +
"|  " +
"|  |") %
(dir_name,theotherdir,'gt',nameonly,
dir_name,subdir,methods[0],nameonly,
dir_name,subdir,methods[1],nameonly,
dir_name,subdir,methods[2],nameonly,
dir_name,subdir,methods[3],nameonly,
dir_name,subdir,methods[4],nameonly,
dir_name,subdir,methods[5],nameonly,
dir_name,subdir,'gt',nameonly,
))