Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-11-09 23:12:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: splitting true images in two

Browse source 
Former-commit-id: ad2d1d854f331672afa8f5443952151d8978e684 [formerly 79566cff12dad5a2ca4ee9b758fb7da98f652a59]
Former-commit-id: 5406367c9bdc195310b505cbcb90cbec1838f02c
This commit is contained in:
Laurent Fainsin 2022-07-08 11:34:50 +02:00
parent 164417b13f
commit 6acac70972
2 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/train.py
View file

@ -10,9 +10,8 @@ from unet import UNet
CONFIG = { CONFIG = {
"DIR_TRAIN_IMG": "/home/lilian/data_disk/lfainsin/train/", "DIR_TRAIN_IMG": "/home/lilian/data_disk/lfainsin/train/",
"DIR_VALID_IMG": "/home/lilian/data_disk/lfainsin/val/", "DIR_VALID_IMG": "//home/lilian/data_disk/lfainsin/test_split/",
"DIR_TEST_IMG": "/home/lilian/data_disk/lfainsin/test/", "DIR_SPHERE": "/home/lilian/data_disk/lfainsin/spheres+real_split/",
"DIR_SPHERE": "/home/lilian/data_disk/lfainsin/realspheres/",
"FEATURES": [8, 16, 32, 64], "FEATURES": [8, 16, 32, 64],
"N_CHANNELS": 3, "N_CHANNELS": 3,
"N_CLASSES": 1, "N_CLASSES": 1,

2
src/unet/model.py
View file

@ -93,7 +93,7 @@ class UNet(pl.LightningModule):
) )
def val_dataloader(self): 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( return DataLoader(
ds_valid, ds_valid,