From 6acac709721e19604c34512248f714d1485883f0 Mon Sep 17 00:00:00 2001 From: Laurent Fainsin Date: Fri, 8 Jul 2022 11:34:50 +0200 Subject: [PATCH] feat: splitting true images in two Former-commit-id: ad2d1d854f331672afa8f5443952151d8978e684 [formerly 79566cff12dad5a2ca4ee9b758fb7da98f652a59] Former-commit-id: 5406367c9bdc195310b505cbcb90cbec1838f02c --- src/train.py | 5 ++--- src/unet/model.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/train.py b/src/train.py index a1e54c8..84ccb87 100644 --- a/src/train.py +++ b/src/train.py @@ -10,9 +10,8 @@ from unet import UNet CONFIG = { "DIR_TRAIN_IMG": "/home/lilian/data_disk/lfainsin/train/", - "DIR_VALID_IMG": "/home/lilian/data_disk/lfainsin/val/", - "DIR_TEST_IMG": "/home/lilian/data_disk/lfainsin/test/", - "DIR_SPHERE": "/home/lilian/data_disk/lfainsin/realspheres/", + "DIR_VALID_IMG": "//home/lilian/data_disk/lfainsin/test_split/", + "DIR_SPHERE": "/home/lilian/data_disk/lfainsin/spheres+real_split/", "FEATURES": [8, 16, 32, 64], "N_CHANNELS": 3, "N_CLASSES": 1, diff --git a/src/unet/model.py b/src/unet/model.py index c244735..93872f5 100644 --- a/src/unet/model.py +++ b/src/unet/model.py @@ -93,7 +93,7 @@ class UNet(pl.LightningModule): ) def val_dataloader(self): - ds_valid = SphereDataset(image_dir=wandb.config.DIR_TEST_IMG) + ds_valid = SphereDataset(image_dir=wandb.config.DIR_VALID_IMG) return DataLoader( ds_valid,