From 37abbc78fa3f13a97a613dfb18de88da9663b3cc Mon Sep 17 00:00:00 2001
From: elodiegermani1 <elodie.germani@irisa.fr>
Date: Tue, 12 Mar 2024 16:17:50 +0100
Subject: [PATCH] CycleGAN
---
src/training.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/training.py b/src/training.py
index ad04cbf..aa212c7 100755
--- a/src/training.py
+++ b/src/training.py
@@ -46,6 +46,7 @@ def main():
print('Number of images in VALID:', len(valid_dataset.data_B))
elif dataset_type == 'unpaired':
+ df_file = f'{dpath}/data/train-dataset_rh_4class-jeanzay.csv'
train_dataset = ds.UnpairedImageDataset(subset_A, subset_B, df_file, contrast_list)
print('Number of images in TRAIN:', len(train_dataset.data))
--
GitLab