The evaluation metric for this experiment was used from (TorchMetrics)
-
First of all, according to ·requirement.txt· install dependent library
-
use
LunaPrepare or LIDCPrepare
can parpare training data -
use
trainLIDC or trainLuna
can train different models -
use
evaluateLIDC
orevaluateLuna
can evaluate different models
Note: Note that the segmentation models used in this project are all from github