基于pytorch的预训练模型(resnet34)到ONNX的图像识别,本地部署摄像头、视频识别

目录

1. 介绍

2. main 函数

3. utils

process_frame 函数

4. 代码


1. 介绍

本内容使用的网络是pytorch提供在ImageNet的预训练模型resnet34

关于预训练模型如何转换成 ONNX 格式,参考这个:pytorch模型转ONNX

下载地址:

猜你喜欢

转载自blog.csdn.net/qq_44886601/article/details/130725134