Retinaface Pytorch转ONNX

以你之姓@ 2023-01-16 11:09 143阅读 0赞

环境：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaWNhbzE5OTA_size_16_color_FFFFFF_t_70][]

Retinaface Pytorch源码：

import time
    import torch
    import torch.nn as nn
    import torchvision.models._utils as _utils
    import torchvision.models as models
    import torch.nn.functional as F
    from torch.autograd import Variable
    
    
    def conv_bn(inp, oup, stride = 1, leaky = 0):
        return nn.Sequential(
            nn.Conv2d(inp, oup, 3, stride, 1, bias=False),
            nn.BatchNorm2d(oup),
            nn.LeakyReLU(negative_slope=leaky, inplace=True)
        )
    
    def conv_bn_no_relu(inp, oup, stride):
        return nn.Sequential(
            nn.Conv2d(inp, oup, 3, stride, 1, bias=False),
            nn.BatchNorm2d(oup),
        )
    
    def conv_bn1X1(inp, oup, stride, leaky=0):
        return nn.Sequential(
            nn.

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaWNhbzE5OTA_size_16_color_FFFFFF_t_70]: /images/20221022/2f245730fcb34badaff1342685f26c89.png

发表评论取消回复

表情：

评论列表（有 0 条评论，143人围观）

还没有评论，来说两句吧...

相关阅读

相关 Retinaface Pytorch转ONNX

环境： ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3d

以你之姓@/ 2023年01月16日 11:09/ 0 赞/ 144 阅读

相关 pytorch转换onnx,测试onnx,pytorch 模型结果是否一致

def to_onnx(): dummy_input = torch.randn(1, 3, 112, 112, dtype=torch.float)

小灰灰/ 2023年01月03日 04:16/ 0 赞/ 414 阅读

相关 PyTorch - torch.onnx

PyTorch - torch.onnx Table of Contents (目录) PyTorch Tutorials [https://pytorch.org

灰太狼/ 2022年12月23日 03:28/ 0 赞/ 196 阅读

相关 pytorch pth 模型转 onnx模型，并验证结果正确性

> pytorch 模型部署很重要的一步是转存pth模型为ONNX，本文记录方法。转存 onnx 建立自己的pytorch模型，并加载权重 model

刺骨的言语ヽ痛彻心扉/ 2022年12月23日 00:37/ 0 赞/ 437 阅读

相关 pytorch模型转ONNX模型

转载请注明作者和出处： [http://blog.csdn.net/john\_bh/][http_blog.csdn.net_john_bh] 文章目录

刺骨的言语ヽ痛彻心扉/ 2022年11月29日 05:56/ 0 赞/ 227 阅读

相关【人脸检测】复现Pytorch_Retinaface(Pytorch版本)

[https://github.com/biubug6/Pytorch\_Retinaface][https_github.com_biubug6_Pytorch_Retina

淩亂°似流年/ 2022年11月21日 14:58/ 0 赞/ 277 阅读

相关 onnx2pytorch和onnx-simplifer新版介绍

【GiantPandaCV导语】本文是ONNX2Pytorch思路分享以及onnx-simplifier新版简要介绍。ONNX2Pytorch工具已经测试了onnx model

旧城等待，/ 2022年11月03日 11:57/ 0 赞/ 348 阅读

相关 0041-pytorch入门-猫狗二分类-pth转onnx模型

-- coding: UTF-8 -- Created by -牧野- CSDN https://blog.csdn.net/dcrmg/article/d

柔光的暖阳◎/ 2022年10月24日 01:29/ 0 赞/ 413 阅读

相关 Pytorch模型(.pth)转onnx模型(.onnx)

简介 Open Neural Network Exchange（ONNX，开放神经网络交换）格式，是一个用于表示深度学习模型的标准，可使模型在不同框架之间进行转移。 ON

Myth丶恋晨/ 2022年09月11日 02:25/ 0 赞/ 352 阅读

相关 Pytorch模型转onnx、caffe、TorchScript通用代码，onnx模型推理

[https://blog.csdn.net/zengwubbb/article/details/108696616?spm=1001.2014.3001.5501][http

待我称王封你为后i/ 2022年09月10日 12:22/ 0 赞/ 252 阅读