27 lines
762 B
Python
27 lines
762 B
Python
import random
|
|
from pathlib import Path
|
|
|
|
DATA = sorted(Path("/mnt/wsl/PHYSICALDRIVE1/data/unsplash").glob("*.jpg"))
|
|
TRAIN_SIZE = 0.9
|
|
|
|
CACHE_PATH = Path("/mnt/wsl/PHYSICALDRIVE1/data/cache2")
|
|
MODELS_PATH = Path("/home/andras/projects/bipolaroid/models")
|
|
LOGS_PATH = Path("/home/andras/projects/bipolaroid/logs")
|
|
RUNS_PATH = Path("/home/andras/projects/bipolaroid/runs")
|
|
|
|
|
|
for path in [CACHE_PATH, MODELS_PATH, LOGS_PATH, RUNS_PATH]:
|
|
path.mkdir(exist_ok=True, parents=True)
|
|
|
|
|
|
length = len(DATA)
|
|
indices = list(range(length))
|
|
|
|
random.seed(42)
|
|
random.shuffle(indices)
|
|
|
|
train_indices = indices[: int(length * TRAIN_SIZE)]
|
|
test_indices = indices[int(length * TRAIN_SIZE) :]
|
|
|
|
TRAIN_DATA = [DATA[i] for i in train_indices]
|
|
TEST_DATA = [DATA[i] for i in test_indices]
|