YOLOv5车牌+关键点检测

目录

0. 项目简介

1. 数据处理

2. yolov5配置部分

3. 训练

4. 验证

5. 导出onnx

6. 业务流测试（检测+矫正+识别）

7. 车牌关键点说明

7.1 关键点检测分支

7.2 损失函数

8. yolov5源码修改详情

8.1 数据加载 + 增广

8.2 检测头修改

8.3 Loss

8.4 验证、日志

9. 总结

---

In [1]

!pip install albumentations -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

Looking in indexes: http://mirrors.aliyun.com/pypi/simple/
Collecting albumentations
  Downloading http://mirrors.aliyun.com/pypi/packages/9b/f6/c486cedb4f75147232f32ec4c97026714cfef7c7e247a1f0427bc5489f66/albumentations-1.3.1-py3-none-any.whl (125 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 125.7/125.7 kB 1.3 MB/s eta 0:00:00a 0:00:01
Requirement already satisfied: numpy>=1.11.1 in /opt/conda/envs/python35-paddle120-env/lib/python3.9/site-packages (from albumentations) (1.23.5)
Collecting opencv-python-headless>=4.1.1
  Downloading http://mirrors.aliyun.com/pypi