/Dual-Stream-PRNet-Plus

code for Dual-Stream PRNet++

Primary LanguagePython

Dual-Stream-PRNet-Plus

Official Code for Dual-stream Pyramid Registration Network（Dual-Stream PRNet++ ）Published in Medical Image Analysis

build correlation_cuda

cuda==10.1, torch=1.40 cd ./correlation_package
rm -rf *_cuda.egg-info build dist pycache
python3 setup.py install --user

Training

CUDA_VISIBLE_DEVICES=4,5 python3 launch.py --master_port 29678 train_lpba40.py --net voxnet_fusion --train_data_root ./data/LPBA40/delineation_space

Testing

python3 test_lpba.py --net voxnet_fusion --data_root ./data/LPBA40/test/ --model_file ./models/lpba40_cascade_best.pt