Skip to content
Snippets Groups Projects
Commit 3e3382b7 authored by maali's avatar maali
Browse files

Merge branch 'dev' of https://gitlab.liris.cnrs.fr/maali/pvnet_method into dev

parents 8dc0b495 f9443164
No related branches found
No related tags found
No related merge requests found
...@@ -330,8 +330,9 @@ def getDataSplit(genNew=False, split=.8, modelClass='cat'): ...@@ -330,8 +330,9 @@ def getDataSplit(genNew=False, split=.8, modelClass='cat'):
def getDataSplit_Fruits(genNew=True, 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 # access training data, get jpeg, mask, label filenames split into training / validation sets
if genNew: # create split 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 basePathTraining = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Training/' + modelClass
basePathEvaluating = f'/home/mahmoud/pvnet_method/Generated_Worlds_/Generated_Worlds_Evaluating/{modelClass}' basePathEvaluating = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Evaluating/' + modelClass
masterList_Training = getMasterList_Fruits(basePathTraining) masterList_Training = getMasterList_Fruits(basePathTraining)
masterList_Evalution = getMasterList_Fruits(basePathEvaluating) masterList_Evalution = getMasterList_Fruits(basePathEvaluating)
random.shuffle(masterList_Training) random.shuffle(masterList_Training)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment