import os
import numpy as np
import random
from PolarVW.loader import DBLoader
from PolarVW.db import adddb
from PolarVW.variables import cdir
dbnames = ['capricebaseline',
'careiicarotid',
'careiicarotidcbir',
'kowa']
dbloader = DBLoader()
for dbname in dbnames:
adddb(dbloader,dbname)
tot_size=1157#len(list(set([i[:i.find('E')] for i in case_raw])))
case_unique_filename = cdir+'/case_uniq'+str(tot_size)+'.npy'
if not os.path.exists(case_unique_filename):
case_uniq = list(set([i['casename'] for i in dbloader.pilist]))
random.shuffle(case_uniq)
np.save('case_uniq'+str(tot_size), case_uniq)
print('generate','case_uniq'+str(tot_size))
case_uniq = np.load(case_unique_filename)
print('loaded', case_unique_filename, case_uniq.shape)
trainsp = int(0.8 * len(case_uniq))
valsp = int(0.9 * len(case_uniq))
train_caselist = case_uniq[:trainsp]
val_caselist = case_uniq[trainsp:valsp]
test_caselist = case_uniq[valsp:]
import sys
sys.path.append(r'\\DESKTOP4\Dtensorflow\LiChen\VW\PolarReg')
from PolarVW.loader import DBLoader, CaseLoader
from PolarVW.variables import DESKTOPdir, DATADESKTOPdir, taskdir
from PolarVW.db import adddb
taskname = 'PolarReg1-5'
if not os.path.exists(taskdir + '/' + taskname):
os.mkdir(taskdir + '/' + taskname)
print('create new folder ' + taskdir + '/' + taskname)
patchdir = os.path.join(DATADESKTOPdir, 'DVWIMAGES/casepatch/Carotid')
dbnames = ['capricebaseline', 'careiicarotid', 'careiicarotidcbir', 'kowa']
dbloader = DBLoader()
for dbname in dbnames:
adddb(dbloader, dbname)
cfg = {}
cfg['width'] = 512 // 2
cfg['height'] = 512 // 2
cfg['patchheight'] = 256
cfg['depth'] = 3
cfg['channel'] = 1
cfg['G'] = len(get_available_gpus())
cfg['taskdir'] = taskdir
cfg['taskname'] = taskname
cfg['batchsize'] = 16
cfg['rottimes'] = 4
cfg['startepoch'] = 232
if platform == 'win32':
DESKTOPdir = '//Desktop4'
DATADESKTOPdir = '//Desktop2'
else:
# ubuntu
DESKTOPdir = '/mnt/desktop4'
DATADESKTOPdir = 'mnt/desktop2'
taskdir = os.path.join(DESKTOPdir, 'Dtensorflow/LiChen/VW/')
if not os.path.exists(taskdir + '/' + taskname):
os.mkdir(taskdir + '/' + taskname)
print('create new folder ' + taskdir + '/' + taskname)
patchdir = os.path.join(DATADESKTOPdir, 'DVWIMAGES/casepatch/Carotid')
dbnames = ['capricebaseline',
'careiicarotid',
'careiicarotidcbir',
'kowa']
dbloader = DBLoader()
for dbname in dbnames:
dbloader.adddb(dbname, patchdir)
cfg = {}
cfg['width'] = 256
cfg['height'] = 256
cfg['depth'] = 3
cfg['channel'] = 1
cfg['batch_size'] = 512
cfg['G'] = len(get_available_gpus())