提交 11205a15 编写于 作者: X xiaoting 提交者: Kaipeng Deng

polish yolo readme (#2201)

上级 21162a2a
......@@ -161,7 +161,7 @@ Inference is used to get prediction score or image features based on trained mod
--weights=${path_to_weights} \
--image_path=data/COCO17/val2017/ \
--image_name=000000000139.jpg \
--draw_threshold=0.5
--draw_thresh=0.5
- Set ```export CUDA_VISIBLE_DEVICES=0``` to specifiy one GPU to infer.
......
......@@ -163,7 +163,7 @@ Train Loss
--weights=${path_to_weights} \
--image_path=data/COCO17/val2017/ \
--image_name=000000000139.jpg \
--draw_threshold=0.5
--draw_thresh=0.5
- 通过设置export CUDA\_VISIBLE\_DEVICES=0指定单卡GPU预测。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册