From 264d8a014c8b694399e5b727d5614d73d6d97960 Mon Sep 17 00:00:00 2001 From: "wenfeng.zhang" Date: Wed, 31 Jan 2024 11:51:52 +0800 Subject: [PATCH] add yolov5 model use paddlepaddle framework --- cv/detection/yolov5/paddlepaddle/README.md | 35 ++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 cv/detection/yolov5/paddlepaddle/README.md diff --git a/cv/detection/yolov5/paddlepaddle/README.md b/cv/detection/yolov5/paddlepaddle/README.md new file mode 100644 index 000000000..bcec178ad --- /dev/null +++ b/cv/detection/yolov5/paddlepaddle/README.md @@ -0,0 +1,35 @@ +# YOLOv5 + +## Model description + +WOLOv5 🚀 is the world's most loved vision AI, representing Ultralytics open-source research into future vision AI methods, incorporating lessons learned and best practices evolved over thousands of hours of research and development. + +We hope that the resources here will help you get the most out of YOLOv5. Please browse the YOLOv5 Docs for details, raise an issue on GitHub for support, and join our Discord community for questions and discussions! + +To request an Enterprise License please complete the form at [Ultralytics Licensing](https://ultralytics.com/license). + +## Step 1: Installing + +```bash +git clone --recursive -b release/2.6 https://github.com/PaddlePaddle/PaddleYOLO.git +cd PaddleYOLO +pip3 install -r requirements.txt # install +``` + +## Step 2: Download data + +```bash +python3 dataset/coco/download_coco.py +``` + +## Step 3: Run YOLOv5 + +```bash +# Make sure your dataset path is the same as above +cd PaddleDetection +export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 +python3 -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/yolov5/yolov5_s_300e_coco.yml --eval +``` + +## Reference +- [PaddleYOLO](https://github.com/PaddlePaddle/PaddleYOLO) \ No newline at end of file -- Gitee