From f94431643f17ac85b59b49eef9c4e370e1674a58 Mon Sep 17 00:00:00 2001
From: Gduret <guillaume.duret@ec-lyon.fr>
Date: Mon, 13 Mar 2023 16:17:52 +0100
Subject: [PATCH] minor path issue
---
data.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/data.py b/data.py
index e23af56..ffde80e 100644
--- a/data.py
+++ b/data.py
@@ -310,8 +310,9 @@ def getDataSplit(genNew=False, split=.8, modelClass='cat'):
def getDataSplit_Fruits(genNew=True, split=.8, modelClass='cat'):
# access training data, get jpeg, mask, label filenames split into training / validation sets
if genNew: # create split
- basePathTraining = f'/home/mahmoud/pvnet_method/Generated_Worlds_/Generated_Worlds_Training/{modelClass}' # os.path.dirname(os.path.realpath(__file__)) + '/LINEMOD/' + modelClass
- basePathEvaluating = f'/home/mahmoud/pvnet_method/Generated_Worlds_/Generated_Worlds_Evaluating/{modelClass}'
+ basePathTraining = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Training/' + modelClass
+ basePathEvaluating = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Evaluating/' + modelClass
+
masterList_Training = getMasterList_Fruits(basePathTraining)
masterList_Evalution = getMasterList_Fruits(basePathEvaluating)
random.shuffle(masterList_Training)
--
GitLab