Skip to content

Latest commit

 

History

History
33 lines (31 loc) · 805 Bytes

README.md

File metadata and controls

33 lines (31 loc) · 805 Bytes
Raw

4DCT-OpFlow

4D-CT Dicom Images Optical Flow Neural Network

Train without cost unrolling

  • Branch out from gpu branch
  • Run the following command: 
    ~/4DCT-OpFlow/
❯ python main_train.py -c configs/server.json
  • Run from checkpoint: 
    ~/4DCT-OpFlow/
❯ python main_train.py -c configs/server.json -l checkpoints/some_checkpoint.tar.gz

Train with cost unrolling

  • Branch out from cost_unrolling_gpu branch
  • Run the following command: 
    ~/4DCT-OpFlow/
❯ python main_train.py -c configs/server_admm_v2.json
  • Run from checkpoint: 
    ~/4DCT-OpFlow/
❯ python main_train.py -c configs/server_admm_v2.json -l checkpoints/some_checkpoint.tar.gz

For more info on training

 ~/4DCT-OpFlow/
 ❯ python main_train.py --help