青光眼那个用的是BinaryResNet2dModel？

junqiangchen / PytorchDeepLearing

Meidcal Image Segmentation Pytorch Version

216 stars 38 forks source link

青光眼那个用的是BinaryResNet2dModel？ #13

Closed wentixiaogege closed 7 months ago

wentixiaogege commented 8 months ago

这个配置么？TASK1

resnet2d = BinaryResNet2dModel(image_height=1024, image_width=1024, image_channel=1, numclass=1,
                              batch_size=32, loss_name='BinaryCrossEntropyLoss',use_cuda=use_cuda)
resnet2d.trainprocess(trainimages, trainlabels, valimages, vallabels, model_dir='log/BinaryResNet2d/CE', epochs=300,
                      lr=0.005)

junqiangchen commented 8 months ago

@wentixiaogege 可以的，也可以用MutilResNet2dModel