numpy
scipy
matplotlib
tqdm
scikit-learn
scikit-image
pillow
SimpleITK
tensorflow==1.15.0
keras==2.1.6
h5py==2.10.0
protobuf==3.20.*
opencv-python
根據(jù)以上環(huán)境配置
修改segcaps中的:load_3D_data
第40行修改為:with open(join(root, 'split_lists', 'train_split_' +str(split) +'.csv'), 'r')as f:
第44行修改為:with open(join(root, 'split_lists', 'test_split_' +str(split) +'.csv'), 'r')as f:
第105行修改為:with open(join(outdir,'train_split_' +str(n) +'.csv'), 'w')as csvfile:
第109行修改為:with open(join(outdir,'test_split_' +str(n) +'.csv'), 'w')as csvfile:
280行增加(此為迭代器內(nèi)容):def __next__(self):
with self.lock:
return next(self.it)
train.py的114行修改為:with open(join(output_dir, args.save_prefix + outfile +'scores.csv'), 'w')as csvfile: