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

最新推荐文章于 2023-12-14 22:07:39 发布
最新推荐文章于 2023-12-14 22:07:39 发布
阅读量1.2k 收藏
点赞数
分类专栏： 深度学习之应用篇 文章标签： pytorch 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_44886601/article/details/130725134
版权
本博客介绍了如何使用预训练的PyTorch ResNet34模型将其转换为ONNX格式，以便进行本地摄像头和视频的图像识别。内容包括模型转换、主函数和辅助工具函数的详细步骤，以及处理单帧图像的流程，最终展示识别结果并计算FPS。
摘要由CSDN通过智能技术生成

目录

1. 介绍

2. main 函数

3. utils

process_frame 函数

4. 代码

1. 介绍

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

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

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

项目的文件目录如下：

  • 其中labels文件是ImageNet 的label文件，以及对应的中文label信息
  • to_onnx 中是将pytorch模型转为ONNX文件，并检测完整性
  • main函数就是识别的主函数，utils 是相关的工具函数

了解本专栏 订阅专栏 解锁全文 超级会员免费看
听风吹等浪起
关注
专栏目录
订阅专栏
基于pytorch的预训练模型转换为ONNX的图像识别
Henry的博客
05-16 759
将预训练模型resnet转为通用格式ONNX，并对单张图像进行识别
多模态大模型：技术原理与实战 在LLM时代，对软件研发的更多思考————从软件 1.0 迈向软件 2.0 时代
最新发布
程序员光剑
06-29 791
软件1.0 vs 软件2.0 - **软件1.0**：传统的软件开发方法，通过人工编写明确的**程序逻辑和规则**来实现功能。 - **软件2.0**：利用AI和机器学习技术，通过**训练模型来"学习"如何执行任务，而不是显式编程**。在这种范式下，软件的行为更多地**由数据和学习算法决定，而不是固定的规则。**
resnet各种预训练模型
11-09
resnet预训练模型有resnet18.caffemodel，resnet50.caffemodel，resnet101.caffemodel，resnet152.caffemodel
Pytorch学习笔记（I）——预训练模型（八）：ResNet34网络结构
lockonlxf的博客
05-21 6286
ResNet( (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (relu): ReLU(...
本地下载预训练模型(涉及内容：Resnet等预训练模型地址，以resnet34为例下载预训练模型并移动到指定地址)
分享计算机视觉，C++，网络摄像头研发，音视频开发，嵌入式等知识。
12-01 2340
复制需要下载的模型地址，粘贴到浏览器地址栏中下载，各种模型的下载地址如下：
Pytorch学习笔记（I）——预训练模型（二）：修改网络结构（ResNet34及以下）
lockonlxf的博客
07-05 8437
（pytorch1.0）最近在研究pytorch如何修改与训练模型的网络结构，然后发现了两种版本，一种是细调版，一种是快速版 　　经过一番钻研后发现细调版适合对网络模型进行大幅度的改动（如在原有的结构上穿插着增减层），而快速版适合直接对网络末端的层进行增减。 　　虽然快速版简单易懂，但是还是要对细调版有所了解才能比较，万一以后用的上呢。因此，我就好好研究了一番细调版，结果发现网上的代码或者博客基...
利用Pytorch预训练模型进行图像分类
AIHUBEI的博客
12-14 861
利用预训练模型进行图像分类。
Android上部署pytorch实现拍照性别识别
04-24
通常，这样的模型基于卷积神经网络（CNN），如ResNet、VGG或MobileNet等，它们在大量标注数据上训练，能从面部特征中识别出性别。在本项目中，开发者可能已经将一个预先训练好的模型转换为适合在Android上运行的格式...
pytorch forward_使用OpenVINO加速Pytorch表情识别模型
weixin_39785970的博客
11-21 260
点击上方蓝字关注我们微信公众号：OpenCV学堂关注获取更多计算机视觉与深度学习知识关于模型OpenVINO自带的表情识别模型是Caffe版本的，这里使用的模型是前面一篇文章中训练生成的pytorch全卷积网络，模型基于残差网络结构全卷积分类网络。输入格式：NCHW=1x3x64x64输出格式：NCHW=1x8x1x1支持八种表情识别，列表如下：["neutral","anger","d...
pytorch训练imagenet分类的方法
12-24
1、imagenet数据准备： a、下载数据集 b、提取training data: mkdir train && mv ILSVRC2012_img_train.tar train/ && cd train tar -xvf ILSVRC2012_img_train.tar && rm -f ILSVRC2012_img_train.tar find . -name "*.tar" | while read NAME ; do mkdir -p "${NAME%.tar}"; tar -xvf "${NAME}" -C "${NAME%.tar}"; rm -f "${NAME}"; do
用pytorch训练图像分类器模型导出ONNX测试
牧野的博客
10-29 1万+
项目实现功能 1. 搭建一个简单的图片分类器，完成训练和测试 2. 转换pytorch的pth模型到ONNX格式，加载ONNX并测试 项目结构 images目录下存放训练和测试数据集，本例使用了kaggle竞赛的猫狗数据集，统一resize到了120*120大小；TrainTestConvertOnnx.py 是训练和测试代码，包括了pth模型到onnx的转换。训练在CPU和GP...
pytorch从预训练模型提取图像特征计算featureless
没错我是程序猿
03-05 1529
1.背景描述 想借助一个预训练好的网络（非集成好的）计算feature-loss，预训练网络地址：表情识别net 2具体实操 2.1 加载模型 作者已经给出了预训练好的模型参数和模型代码，首先我们要把模型load进来： from Expression.VGG import VGG model = VGG('VGG19') #check_pth 从网站上download下来P...
基于pytorch的图像识别基础完整教程
热门推荐
m0_60681041的博客
02-12 2万+
基于pytorch的深度学习图像识别基础完整教程 以常见盆栽植物的图像识别示例来驱动学习，通过这个教程，你可以学会深度学习中的图像识别的完整操作并且可以通过这个示例训练出其他的图像识别模型。
4. 使用预训练的PyTorch网络进行图像分类
qq_40985985的博客
01-14 1705
这篇博客将介绍如何使用PyTorch预先训练的网络执行图像分类。利用这些网络只需几行代码就可以准确地对1000个常见对象类别进行分类。这些图像分类网络是开创性的、最先进的图像分类网络，包括VGG16、VGG19、Inception、DenseNet和ResNet。 图像分类允许为输入图像指定一个或多个标签，然而它并没有告诉对象在图像中的位置。要确定给定对象在输入图像中的位置，需要应用对象检测。 下一篇博客将介绍如何使用PyTorch使用专门的对象检测网络检测图像中的对象。.........
5. 使用PyTorch预先训练的模型执行目标检测
qq_40985985的博客
06-28 3868
这篇博客将介绍如何使用PyTorch预训练的网络执行目标检测，这些网络是开创性的、最先进的图像分类网络，包括使用ResNet的更快R-CNN、使用MobileNet的更快R-CNN和RetinaNet。 - 具有ResNet50主干的更快R-CNN（更精确，但速度较慢） - 具有MobileNet主干的更快R-CNN（速度更快，但准确度稍低） - 具有ResNet50主干的RetinaNet（速度和准确性之间的良好平衡）............
Pytorch分类模型转onnx以及onnx模型推理
xys430381_1的专栏
03-28 5103
Pytorch分类模型转onnx以及onnx模型推理 将MobileNetV2的pytorch模型转换为ONNX模型,并用onnxruntime进行测试_cumtchw-程序员秘密 Onnx模型推理（Python）,onnx,python
PyTorch项目笔记（三）使用ImageNet预训练ResNet18模型训练图像分类模型
a1016249126的博客
03-19 1万+
本文介绍了如何在ImageNet预训练模型ResNet18的基础上微调训练自己的图像分类模型
OpenCV dnn模块 分类模型Resnet50 OpenCV dnn模块部署 .onnx模型
zhulong1984的博客
08-31 1793
项目需要，需要基于opencv的dnn模块来调用resnet模型，需要resnet模型先训练，之后再导出onnx。
基于pytorch实现的声纹识别预训练模型(v1.0)
10-31
基于PyTorch实现的声纹识别预训练模型(v1.0)是一个已经在声纹识别领域进行预训练的模型版本。声纹识别是一种通过声音信号来识别个体身份的技术。 这个预训练模型(v1.0)的核心是使用PyTorch框架搭建的神经网络模型。...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

听风吹等浪起

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值