Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
YongtaoGe committed Apr 9, 2024
1 parent e702680 commit 5aa75c8
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 1 deletion.
7 changes: 6 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,12 @@ Download the pre-trained depth model ```depth_v1.zip``` from [BaiduNetDisk](http

Then, place images in the ```./input/``` dictionary, and run the following script. The output depth will be saved in ```./output/```.
```bash
source scripts/inference_depth.sh
sh scripts/inference_depth.sh
```

For surface normal estimation, run the following script:
```bash
bash scripts/inference_normal.sh
```

Thanks to our one-step perception paradigm, the inference process runs much faster. (Around 0.4s for each image on an A800 GPU card.)
Expand Down
9 changes: 9 additions & 0 deletions scripts/inference_normal.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
CUDA_VISIBLE_DEVICES='1'
export PYTHONPATH=./

python tools/inference_genpercept.py \
--input_rgb_dir 'input' \
--output_dir 'output' \
--mode 'normal' \
--checkpoint "weights/normal_v1"

0 comments on commit 5aa75c8

Please sign in to comment.