Skip to content

Commit 3267847

Browse files
committed
better output dirs
1 parent eaf4f7a commit 3267847

File tree

2 files changed

+15
-2
lines changed

2 files changed

+15
-2
lines changed

src/maps.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ def __init__(self, filename, mapName):
1818
data = f.read().split()
1919
self.size = (int(data[0]), int(data[1]))
2020
self.data = [[int(data[i * self.size[1] + j + 2]) for j in range(self.size[1])] for i in range(self.size[0])]
21+
logging.info(f"Your map is {self.size[0]}x{self.size[1]}")
2122

2223
# Converter object that can return character for OpenStreetMap layer group
2324
# Credit - Bahar
@@ -80,7 +81,7 @@ def generate_patches(self, mapReader, image_groups=3, outDirectory=None):
8081
image_groups (int, optional): _description_. Defaults to 3.
8182
"""
8283
if outDirectory is None:
83-
outDirectory = os.path.join(self.outputDir, mapReader.mapName, str(self.window_size))
84+
outDirectory = os.path.join(self.outputDir, mapReader.mapName, f"{self.window_size[0]}x{self.window_size[1]}", f"{self.step_size}x{self.step_size}")
8485
os.makedirs(outDirectory, exist_ok=True)
8586

8687
img_group_number = 0

src/run.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,18 @@
11
# Run this script to generate data in /output directory.
2+
import logging
3+
import sys
4+
5+
root = logging.getLogger()
6+
root.setLevel(logging.INFO)
7+
8+
handler = logging.StreamHandler(sys.stdout)
9+
handler.setLevel(logging.INFO)
10+
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
11+
handler.setFormatter(formatter)
12+
root.addHandler(handler)
13+
214
import maps
315

416
sfMap = maps.MapReader('../data/input/sf_layered.txt', "SF_Layered")
5-
mapsDataset = maps.MapsDataset((32, 32), 2, 1280, maps.single_layer_converter) #Third parameter is the group size
17+
mapsDataset = maps.MapsDataset((32, 32), 10, 1280, maps.single_layer_converter) #Third parameter is the group size
618
mapsDataset.generate_patches(sfMap) #This will generate dill files which contain the saved sample lists.

0 commit comments

Comments
 (0)