This is a classification of lung nodules using pytorch, it is the final project of SJTU-EE228.
Link：https://pan.baidu.com/s/1XFreGXYcs8wS-TGxmKvfwg
Extraction code：q5wg
acsconv: a converter that turns a 2D network into a 3D network
densenet: an implementation of 2D Densenet network, needs to be combined with a converter when using it
resnet3d: an implementation of 3D Resnet network
shake_shake: an implementation of Shake_shake network, needs to be combined with a converter when using it
vgg: an implementation of 2D vgg network, needs to be combined with a converter when using it
train_densenet169: a training program for densenet169 model
train_mixupresnet50: a training program for mixup-resnet50 model
train_mixupresnet50: a training program for resnet26 model
train_shake: a training program for shake_shake model
train_vgg: a training program for vgg model
test: load model
You only need to run the test file to get the prediction of the test data. Please write the test data path before running, and the model and network files are in the same folder