提交 30f90d6a 编写于 作者: W wangguanzhong 提交者: GitHub

fix doc related to CPU (#3048)

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