pytorch模型转ONNX模型及ONNX模型推理和可视化

最新推荐文章于 2025-03-06 00:05:06 发布

zjLOVEcyj

最新推荐文章于 2025-03-06 00:05:06 发布

阅读量742

点赞数 1

分类专栏： pytorch 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/cyj5201314/article/details/121118997

版权

pytorch模型转ONNX模型

import torch

class ConNet(torch.nn.Module):
    def __init__(self):
        super(ConNet, self).__init__()
        self.conv1 = torch.nn.Sequential(
            torch.nn.Conv2d(3, 16, 3, 1, 1),
            torch.nn.ReLU(),
            torch.nn.AvgPool2d(2, 2)
        )
        self.conv2 = torch.nn.Sequential(
            torch.nn.Conv2d(16