Update README.md

README.md

@@ -3,12 +3,7 @@ license: apache-2.0
 ---
 # Instructions
 
-### 1-Download
-Download the singularity container.
-```
-wget https://huggingface.co/qicq1c/SuPreM/resolve/main/suprem_final.sif
-```
-### 2-Data preparation
+## 1-Data preparation
 This is how `inputs_data` organizes
 ```
     $inputs_data/
@@ -20,9 +15,27 @@ This is how `inputs_data` organizes
     │   └── ct.nii.gz
     ...
 ```
+## 2-Download
+You can choose to use the docker image (recommand) or the singularity container.
 
-### 3-Inference
+#### Download the docker image.
+```
+docker pull qchen99/suprem:v1
+```
+or
+#### Download the singularity container.
+```
+wget https://huggingface.co/qicq1c/SuPreM/resolve/main/suprem_final.sif
+```
+
+## 3-Inference
 You can directly perform inference on your own data. Simply modify `inputs_data` into your data path and adjust `outputs_data` to specify the desired output location for the segmentation results.
+#### Use Docker
+```
+sudo docker container run --gpus "device=0" -m 128G --rm -v $inputs_data:/workspace/inputs/ -v $outputs_data:/workspace/outputs/ qchen99/suprem:v1 /bin/bash -c "sh predict.sh"
+```
+or
+#### Use Singularity
 ```
 SINGULARITYENV_CUDA_VISIBLE_DEVICES=0 singularity run --nv -B $inputs_data:/workspace/inputs -B $outputs_data:/workspace/outputs suprem_final.sif
 ```