From abae35fc95e9e0d0ce6f7652931f034f72924930 Mon Sep 17 00:00:00 2001
From: "jose.luis.holgado" <jose.l.holgadoalvarez@campus.tu-berlin.de>
Date: Tue, 30 Jun 2020 13:14:21 +0200
Subject: [PATCH] update in test script
---
test.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/test.py b/test.py
index 8d9bef7..f6d41f7 100644
--- a/test.py
+++ b/test.py
@@ -62,7 +62,7 @@ def update_progress(progress):
if __name__ == '__main__':
-
+ opt = TestOptions().parse()
tiles = 9
for tile in range(tiles):
@@ -77,6 +77,7 @@ if __name__ == '__main__':
opt.dataset = opt.dataset.format(tile)
coord_path = os.path.join(opt.dataroot.replace("patches",""),opt.dataset)
+ print(coord_path)
with open(coord_path) as json_file:
coord_dict = json.load(json_file)
opt.num_test = len(np.array(list(coord_dict.keys())))
@@ -110,7 +111,7 @@ if __name__ == '__main__':
origin_x = []
origin_y = []
target_tile = 0
- os.mkdir(path_dir_image)
+ os.makedirs(path_dir_image)
for idx, data in enumerate(dataset):
if idx >= opt.num_test: # only apply our model to opt.num_test images.
--
GitLab