-
Notifications
You must be signed in to change notification settings - Fork 1
/
concatenate.py
63 lines (50 loc) · 2.19 KB
/
concatenate.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
59
60
61
62
63
# Copyright 2018 Giovanni Giacomo
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import glob
import scipy.misc
from matplotlib.pyplot import imread, imsave
import numpy as np
EXECUTION_NAME = "2019-04-14_23:03"
WIDTH = 256
HEIGHT = 128
USE_REAL = False
def main():
data_son = sorted(glob.glob("./datasets/aracati/test/input/*.png"))
data_fak = sorted(glob.glob("./executions/{}/results/*.png".format(EXECUTION_NAME)))
if USE_REAL:
data_rea = sorted(glob.glob("./datasets/aracati/test/gt/*.png"))
if USE_REAL:
assert(len(data_son) == len(data_fak) == len(data_rea))
else:
assert(len(data_son) == len(data_fak))
for i in range(len(data_fak)):
image_son = imread(data_son[i]).astype(np.float)
image_son = scipy.misc.imresize(image_son, [HEIGHT, WIDTH])
image_son = np.asarray(np.dstack((image_son, image_son, image_son)), dtype=np.uint8)
image_fak = imread(data_fak[i]).astype(np.float)
image_fak = scipy.misc.imresize(image_fak, [HEIGHT, WIDTH])
if USE_REAL:
image_rea = imread(data_rea[i]).astype(np.float)
image_rea = scipy.misc.imresize(image_rea, [HEIGHT, WIDTH])
if USE_REAL:
image_res = np.concatenate((image_son[:,:,:3], image_fak[:,:,:3], image_rea[:,:,:3]), axis=1)
else:
image_res = np.concatenate((image_son[:,:,:3], image_fak[:,:,:3]), axis=1)
imsave("./executions/{}/presentations/test_{:05d}.png".format(EXECUTION_NAME, i), image_res)
print("CONCATENATING: Finished Image {:05d}".format(i))
if __name__ == '__main__':
main()