# disc_loss=DisLoss(), vertebra_loss=DisLoss()
)
# 放置到GPU
dis_model.cuda()
# 打印模型参数
print(dis_model)
# 设定训练参数
train_dataloader = DisDataLoader(train_studies,
train_annotation,
batch_size=8,
num_workers=0,
num_rep=20,
prob_rotate=1,
max_angel=180,
sagittal_size=dis_model.sagittal_size,
transverse_size=dis_model.transverse_size,
k_nearest=dis_model.k_nearest,
sagittal_shift=dis_model.sagittal_shift,
pin_memory=False,
sampling_strategy=None)
# 设定验证参数
valid_evaluator = Evaluator(
dis_model,
valid_studies,
'../data/lumbar_train51_annotation.json',
num_rep=5,
max_dist=6,
)
spinal_model=spinal_model)
# 建立疾病分类模型
dis_model = DiseaseModelBase(kp_model, sagittal_size=(512, 512))
# 放置到GPU
dis_model.cuda()
# 打印模型参数
print(dis_model)
# 设定训练参数
train_dataloader = DisDataLoader(train_studies,
train_annotation,
batch_size=8,
num_workers=0,
num_rep=20,
prob_rotate=1,
max_angel=180,
sagittal_size=dis_model.sagittal_size,
transverse_size=dis_model.sagittal_size,
k_nearest=0,
max_dist=6,
sagittal_shift=1,
pin_memory=True)
# 设定验证参数
valid_evaluator = Evaluator(dis_model,
valid_studies,
'../data/lumbar_train51_annotation.json',
num_rep=20,
max_dist=6,
metric='key point recall')