一种鱼眼失真图像的检测方法及系统
Abstract:
本申请提供一种鱼眼失真图像的检测方法及系统。本申请基于PyTorch深度学习框架构建YOLOv5深度学习模型,在YOLOv5深度学习模型输出的训练参数[x,y,w,h,c]中增加有角度信息θ,获得六参数输出的旋转框[x,y,w,h,θ,c],并将YOLOv5深度学习模型中Dataloader模块的维度扩展至六维以对旋转框进行六参数读取,获得适用于鱼眼图像的YOLOv5旋转框模型。本申请的模型经训练后能够通过旋转框算法,在模型内部运算过程中通过旋转框克服鱼眼失真图像中的畸变,兼顾检测精度和检测速度,获得更优检测效果。
Public/Granted literature
Patent Agency Ranking
0/0