|
a |
|
b/(1) PyTorch_HistoNet/params/dirs.py |
|
|
1 |
import os |
|
|
2 |
|
|
|
3 |
|
|
|
4 |
class dirs: |
|
|
5 |
|
|
|
6 |
# init |
|
|
7 |
def __init__(self, dirWorkspaceOrig, dirWorkspaceTest, dbname, csvFile, ROI): |
|
|
8 |
self.dirWorkspaceOrig = dirWorkspaceOrig |
|
|
9 |
self.dirWorkspaceTest = dirWorkspaceTest |
|
|
10 |
self.dbname = dbname |
|
|
11 |
self.ROI = ROI |
|
|
12 |
self.csvFile = csvFile |
|
|
13 |
if not os.path.exists(self.dirWorkspaceTest): |
|
|
14 |
os.mkdir(self.dirWorkspaceTest) |
|
|
15 |
|
|
|
16 |
def computeDirs(self): |
|
|
17 |
dirDbOrig = self.dirWorkspaceOrig + self.dbname + '/' + self.ROI + '/' |
|
|
18 |
dirDbTest = self.dirWorkspaceTest + self.dbname + '_' + self.ROI + '/' |
|
|
19 |
dirOutTrainTest = dirDbTest + 'datastore_trainTest/' |
|
|
20 |
return dirDbOrig, dirDbTest, dirOutTrainTest |
|
|
21 |
|
|
|
22 |
def getCSVfileFull(self): |
|
|
23 |
csvFileFull = self.dirWorkspaceOrig + self.dbname + '/' + self.csvFile |
|
|
24 |
return csvFileFull |
|
|
25 |
|