diff --git a/PaddleCV/yolov3/README.md b/PaddleCV/yolov3/README.md index 6f14bea951a103f3496467f52108db7ab40b8330..109c8dba13ebf51489be3126dd06031d4348cd19 100644 --- a/PaddleCV/yolov3/README.md +++ b/PaddleCV/yolov3/README.md @@ -155,7 +155,9 @@ Inference is used to get prediction score or image features based on trained mod --image_name=000000000139.jpg \ --draw_threshold=0.5 -Inference speed: +- Set ```export CUDA_VISIBLE_DEVICES=0``` to specifiy one GPU to infer. + +Inference speed(Tesla P40): | input size | 608x608 | 416x416 | 320x320 | diff --git a/PaddleCV/yolov3/README_cn.md b/PaddleCV/yolov3/README_cn.md index cd12f8353bb19657931839cbaa51ee983297de17..cfb01ee91d2d8329fd6d747b870807c62864b97a 100644 --- a/PaddleCV/yolov3/README_cn.md +++ b/PaddleCV/yolov3/README_cn.md @@ -157,7 +157,9 @@ Train Loss --image_name=000000000139.jpg \ --draw_threshold=0.5 -模型预测速度: +- 通过设置export CUDA\_VISIBLE\_DEVICES=0指定单卡GPU预测。 + +模型预测速度(Tesla P40): | input size | 608x608 | 416x416 | 320x320 |