diff --git a/test.py b/test.py index 8d9bef784fe873065e0c519590255e8385cda587..f6d41f72b10cb080ca63d0d1aa5c8099d69cc4ad 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.