Commit 90cf3286 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent ca37e0fa
# yolov5_paddle
# yolov5
## 论文
......@@ -90,7 +90,7 @@ COCO2017(在网络良好的情况下,如果没有下载数据集,程序会
```
## 训练
- 训练数据集路径在 config/dataset/coco_detection.yml中修改dataset_dit路径
- 训练数据集路径在 config/dataset/coco_detection.yml中修改dataset_dir路径
- {config}参数使用时应改为配置文件地址: configs/yolov5/yolov5_s_300e_coco.yml [该参数可替换为其他的参数文件地址]
### 单机单卡
......@@ -107,7 +107,7 @@ python -m paddle.distributed.launch tools/train.py -c {config} --amp --eval
export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1
python -m paddle.distributed.launch tools/train.py -c {} --amp --eval
python -m paddle.distributed.launch tools/train.py -c {config} --amp --eval
```
## 推理
......@@ -139,6 +139,6 @@ HIP_VISIBLE_DEVICES=0 python tools/infer.py -c configs/yolov5/yolov5_s_300e_coco
https://developer.hpccube.com/codes/modelzoo/yolov5_paddle
## 参考
## 参考资料
[GitHub - Paddle_yolo](https://github.com/PaddlePaddle/PaddleYOLO)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment