49、Ｒ-YOLOv4旋转目标检测，并进行ncnn和mnn部署开发

基本思想：需要一个简单的旋转目标检测算法，完成业务任务，并输出矫正后的角度信息

onnx建议自己转，云盘中的只支持ncnn的结构被我改了，mnn的onnx是几乎不用改代码的

链接链接: 有偿

第一步：下载台湾小哥的源码

ubuntu@ubuntu:~$ git clone https://github.com/kunnnnethan/R-YOLOv4.git

下载模型，测试一下小哥训练的垃圾模型，注意检测图片的大小为416

测试原图

 测试结果图

第二步：写个pytorch2onnx.py脚本

1）修改一下模型返回值/home/ubuntu/R-YOLOv4/model/yolo.py

return torch.cat([y1, y2, y3], 1), (loss1 + loss2 + loss3)

修改为

return torch.cat([y1, y2, y3], 1)#, (loss1 + loss2 + loss3)

２)然后使用下面的脚本转模型到onnx上

import onnx
import torch
# import trochvision
im