From a98be02f005a52f6354f9f5ef1f66d6d996c1e70 Mon Sep 17 00:00:00 2001
From: Gangwei Xu <gwxu@hust.edu.cn>
Date: Sun, 12 Mar 2023 22:10:49 +0800
Subject: [PATCH] Update README.md

---
 README.md | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index 72e4a37..24d910d 100644
--- a/README.md
+++ b/README.md
@@ -7,6 +7,15 @@ Gangwei Xu, Xianqi Wang, Xiaohuan Ding, Xin Yang<br/>
 
 <img src="IGEV-Stereo/IGEV-Stereo.png">
 
+## Demos
+Pretrained models can be downloaded from [google drive](https://drive.google.com/drive/folders/1SsMHRyN7808jDViMN1sKz1Nx-71JxUuz?usp=share_link)
+
+You can demo a trained model on pairs of images. To predict stereo for Middlebury, run
+```
+python demo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth
+```
+<img src="IGEV-Stereo/demo-imgs.png", width="90%">
+
 ## Environment
 * NVIDIA RTX 3090
 * Python 3.8
@@ -30,15 +39,6 @@ pip install tqdm
 pip install timm==0.5.4
 ```
 
-## Demos
-Pretrained models can be downloaded from [google drive](https://drive.google.com/drive/folders/1SsMHRyN7808jDViMN1sKz1Nx-71JxUuz?usp=share_link)
-
-You can demo a trained model on pairs of images. To predict stereo for Middlebury, run
-```
-python demo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth
-```
-
-
 ## Required Data
 To evaluate/train IGEV-Stereo, you will need to download the required datasets. 
 * [Scene Flow](https://lmb.informatik.uni-freiburg.de/resources/datasets/SceneFlowDatasets.en.html)