Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
lintian-a authored Aug 21, 2024
1 parent 8e8240b commit 3c9ac59
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,15 +22,15 @@ conda create -n nephi python=3.7
pip install -e .
```

## Train the single-res nephi model
## Train the single-res NePhi model
```
python demos/train.py -o=[OUTPUT_FOLDER] -d=[DATASET_FOLDER] -e=[EXPERIMENT_NAME] --train_config=./demos/configs/hybrid_encoder_single_res_reg_full_lung_config.py --epochs_pretrain=2600 --eval_period=-1 --save_period=200 --batch_size=8 -g 0 1 2 3 --lr=1e-4 --lr_step=3000 --with_augmentation=0
```

## Citation
```
@article{tian2023texttt,
title={$$\backslash$texttt $\{$NePhi$\}$ $: Neural Deformation Fields for Approximately Diffeomorphic Medical Image Registration},
title={NePhi: Neural Deformation Fields for Approximately Diffeomorphic Medical Image Registration},
author={Tian, Lin and Greer, Hastings and Est{\'e}par, Ra{\'u}l San Jos{\'e} and Sengupta, Roni and Niethammer, Marc},
journal={arXiv preprint arXiv:2309.07322},
year={2023}
Expand Down

0 comments on commit 3c9ac59

Please sign in to comment.