想做的事情: yolov3的pth模型转成onnx,并用opencv调用原因: 希望不引入深度学习的框架,编译tensorflow谁编译谁懂!而且包文件也会小做法:3. 参考大神的思路: 将yolov3的三个特征图输出重写,输出格式为xyxy,将upsample用转置卷积重写4. yolov3训练的时候用新的网络正常训练,测试的时候np.concatenate三个特征图的输出去预测,这两个结果验证过是一样的5. 想做的事: 将网络输出的pth模型转成onnx, 对onnx模型预测6. 结论:.
torch
2024-01-09
37 点赞
0 评论
56 浏览