From 25ee821c2153a2c184a783f8bcdadb6ebc4e2d81 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 17:13:35 +0200
Subject: [PATCH] fix : typo out file name
---
image_ref/dataset_ref.py | 4 ++--
image_ref/main.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index 37054d71..a086f2d3 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -145,8 +145,8 @@ class ImageFolderDuo(data.Dataset):
imgAER = self.transform(imgAER)
imgANA = self.transform(imgANA)
img_ref = self.transform(img_ref)
- target = 0 if target == label_ref else 1
- return imgAER, imgANA, img_ref, target
+ contrastive_target = 0 if target == label_ref else 1
+ return imgAER, imgANA, img_ref, contrastive_target
def __len__(self):
return len(self.imlist)
diff --git a/image_ref/main.py b/image_ref/main.py
index 704f625d..35fd390c 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -133,7 +133,7 @@ def run_duo(args):
plt.show()
- plt.savefig('output/training_plot_contrastive_{}.png'.format(args.prop))
+ plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
#load and evaluate best model
load_model(model, args.save_path)
--
GitLab