未验证 提交 f8b4dfed 编写于 作者: W wangguanzhong 提交者: GitHub

fix doc related to CPU (#3048)

上级 90b85001
......@@ -28,7 +28,7 @@ python tools/train.py -c configs/faster_rcnn_r50_1x.yml
```bash
export CPU_NUM=8
export PYTHONPATH=$PYTHONPATH:.
python tools/train.py -c configs/faster_rcnn_r50_1x.yml
python tools/train.py -c configs/faster_rcnn_r50_1x.yml -o use_gpu=false
```
##### Optional arguments
......@@ -80,10 +80,9 @@ python -u tools/train.py -c configs/faster_rcnn_r50_1x.yml \
```bash
export CUDA_VISIBLE_DEVICES=0
# run on GPU with:
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
export CUDA_VISIBLE_DEVICES=0
python tools/eval.py -c configs/faster_rcnn_r50_1x.yml
```
......@@ -98,10 +97,9 @@ python tools/eval.py -c configs/faster_rcnn_r50_1x.yml
- configuration options && assign Dataset path
```bash
export CUDA_VISIBLE_DEVICES=0
# run on GPU with:
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
export CUDA_VISIBLE_DEVICES=0
python -u tools/eval.py -c configs/faster_rcnn_r50_1x.yml \
-o weights=output/faster_rcnn_r50_1x/model_final \
-d dataset/coco
......@@ -109,10 +107,9 @@ python -u tools/eval.py -c configs/faster_rcnn_r50_1x.yml \
- Evaluation with json
```bash
export CUDA_VISIBLE_DEVICES=0
# run on GPU with:
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
export CUDA_VISIBLE_DEVICES=0
python tools/eval.py -c configs/faster_rcnn_r50_1x.yml \
--json_eval \
-f evaluation/
......@@ -133,20 +130,18 @@ moment, but it is a planned feature
- Run inference on a single image:
```bash
export CUDA_VISIBLE_DEVICES=0
# run on GPU with:
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
export CUDA_VISIBLE_DEVICES=0
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_img=demo/000000570688.jpg
```
- Multi-image inference:
```bash
export CUDA_VISIBLE_DEVICES=0
# run on GPU with:
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
export CUDA_VISIBLE_DEVICES=0
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_dir=demo
```
......@@ -160,10 +155,9 @@ python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_dir=demo
- Output specified directory && Set up threshold
```bash
export CUDA_VISIBLE_DEVICES=0
# run on GPU with:
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
export CUDA_VISIBLE_DEVICES=0
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml \
--infer_img=demo/000000570688.jpg \
--output_dir=infer_output/ \
......@@ -176,10 +170,9 @@ path, simply add a `--output_dir=` flag.
- Save inference model
```bash
# run on GPU with:
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml \
--infer_img=demo/000000570688.jpg \
--save_inference_model
......
......@@ -29,7 +29,7 @@ python tools/train.py -c configs/faster_rcnn_r50_1x.yml
```bash
export CPU_NUM=8
export PYTHONPATH=$PYTHONPATH:.
python tools/train.py -c configs/faster_rcnn_r50_1x.yml
python tools/train.py -c configs/faster_rcnn_r50_1x.yml -o use_gpu=false
```
##### 可选参数
......@@ -75,10 +75,9 @@ python -u tools/train.py -c configs/faster_rcnn_r50_1x.yml \
```bash
# GPU评估
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# 若使用CPU,则执行
# export CPU_NUM=1
python tools/eval.py -c configs/faster_rcnn_r50_1x.yml
```
......@@ -93,10 +92,9 @@ python tools/eval.py -c configs/faster_rcnn_r50_1x.yml
- 设置配置文件参数 && 指定数据集路径
```bash
# GPU评估
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
python -u tools/eval.py -c configs/faster_rcnn_r50_1x.yml \
-o weights=output/faster_rcnn_r50_1x/model_final \
-d dataset/coco
......@@ -104,10 +102,9 @@ python -u tools/eval.py -c configs/faster_rcnn_r50_1x.yml \
- 通过json文件评估
```bash
# GPU评估
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
python tools/eval.py -c configs/faster_rcnn_r50_1x.yml \
--json_eval \
-f evaluation/
......@@ -127,20 +124,18 @@ json文件必须命名为bbox.json或者mask.json,放在`evaluation/`目录下
- 单图片推断
```bash
# GPU推断
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# 若使用CPU,则执行
# export CPU_NUM=1
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_img=demo/000000570688.jpg
```
- 多图片推断
```bash
# GPU推断
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# 若使用CPU,则执行
# export CPU_NUM=1
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_dir=demo
```
......@@ -154,10 +149,9 @@ python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_dir=demo
- 设置输出路径 && 设置推断阈值
```bash
# GPU推断
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml \
--infer_img=demo/000000570688.jpg \
--output_dir=infer_output/ \
......@@ -171,10 +165,9 @@ python tools/infer.py -c configs/faster_rcnn_r50_1x.yml \
- 保存推断模型
```bash
# GPU推断
export CUDA_VISIBLE_DEVICES=0
export PYTHONPATH=$PYTHONPATH:.
# or run on CPU with:
# export CPU_NUM=1
python tools/infer.py -c configs/faster_rcnn_r50_1x.yml --infer_img=demo/000000570688.jpg \
--save_inference_model
```
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册