diff --git a/data.py b/data.py index e23af56f920968a718ed492e96a4a75d2b6b2121..ffde80ed714080f6b587fac4bb553eeb1f7e18ba 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)