diff --git a/README.md b/README.md index b3214028baff7f1f5f762550bdb294b7cfc58620..e32016cd489ac42729951f8ed7991f8e746129a9 100644 --- a/README.md +++ b/README.md @@ -216,6 +216,7 @@ DeepSparkHub甄选上百个应用算法和模型,覆盖AI和通用计算各领 [YOLOv3](cv/detection/yolov3/pytorch/README.md) | PyTorch | COCO [YOLOv3](cv/detection/yolov3/paddlepaddle/README.md) | PaddlePaddle | COCO [YOLOv3](cv/detection/yolov3/tensorflow/README.md) | TensorFlow | VOC +[YOLOv5](cv/detection/yolov5/paddlepaddle/README.md) | PaddlePaddle | COCO [YOLOv5](cv/detection/yolov5/pytorch/README.md) | PyTorch | COCO [YOLOv6](cv/detection/yolov6/pytorch/README.md)  | PyTorch | COCO [YOLOv7](cv/detection/yolov7/pytorch/README.md) | PyTorch | COCO diff --git a/cv/detection/yolov5/paddlepaddle/README.md b/cv/detection/yolov5/paddlepaddle/README.md new file mode 100644 index 0000000000000000000000000000000000000000..ad81dc9d70ef3020065c16b3b96ab4eba6f0a604 --- /dev/null +++ b/cv/detection/yolov5/paddlepaddle/README.md @@ -0,0 +1,39 @@ +# YOLOv5 + +## Model description + +YOLOv5 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. + +## Step 1: Installation + +```bash +cd deepsparkhub/cv/detection/yolov5/paddlepaddle/ +git clone -b release/2.5 https://github.com/PaddlePaddle/PaddleYOLO.git +cd PaddleYOLO/ +pip3 install -r requirements.txt +python3 setup.py develop +``` + +## Step 2: Preparing datasets + +```bash +python3 dataset/coco/download_coco.py +``` + +## Step 3: Training + +```bash +# Make sure your dataset path is the same as above +export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 +# Select $config_yaml from "configs/yolov5" as you wish +config_yaml=configs/yolov5/yolov5_s_300e_coco.yml +python3 -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c ${config_yaml} --amp --eval +``` + +## Results +GPUs|Model|FPS|ACC +----|---|---|--- +BI-V100 x8|YOLOv5-n| 10.9788 images/s | bbox ap: 0.259 + +## Reference +- [PaddleYOLO](https://github.com/PaddlePaddle/PaddleYOLO)