Commit d8808541 authored by cc215's avatar cc215 💬
Browse files

update readme

parent 0fc11e73
......@@ -13,7 +13,7 @@ author: Chen Chen (cc215@ic.ac.uk)
## Environment
- Python 3.5
- Pytorch 1.0
- Pytorch 1.6 (*please upgrade your pytorch to the latest version, otherwise it may raise error when loading weights from the saved checkpoints.*)
- CUDA(cuda 10.0)
## Dependencies
......@@ -96,7 +96,7 @@ Results will be saved under `test_results` by default
- For details about the data augmentation strategy, please refer to :'dataset_loader/mytransform.py'
## Model update (2021.3.9):
- A model trained on UKBB data (SAX slices) with adversarial data augmentation is available.
- A model trained on UKBB data (SAX slices) with adversarial data augmentation (adversarial noise, adversarial bias field, adversarial morphological deformation, and adversarial affine transformation) is available.
- This model is expected with improved robustness on cross-domain data (especially for images affected by bias field)
- To deploy the model for segmentation, please run the following command to test first:
- run `source ./demo_scripts/predict_test.sh`
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment