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

Update README.md

parent ca37e0fa
# yolov5_paddle # yolov5
## 论文 ## 论文
...@@ -90,7 +90,7 @@ COCO2017(在网络良好的情况下,如果没有下载数据集,程序会 ...@@ -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 [该参数可替换为其他的参数文件地址] - {config}参数使用时应改为配置文件地址: configs/yolov5/yolov5_s_300e_coco.yml [该参数可替换为其他的参数文件地址]
### 单机单卡 ### 单机单卡
...@@ -107,7 +107,7 @@ python -m paddle.distributed.launch tools/train.py -c {config} --amp --eval ...@@ -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 HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export HSA_FORCE_FINE_GRAIN_PCIE=1 export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=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 ...@@ -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 https://developer.hpccube.com/codes/modelzoo/yolov5_paddle
## 参考 ## 参考资料
[GitHub - Paddle_yolo](https://github.com/PaddlePaddle/PaddleYOLO) [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