tensor_proto.raw_data().empty() tensor_proto.float_data().empty() tensor_proto.double_data()解决

应该是上采样

tensor_proto.raw_data().empty() || !tensor_proto.float_data().empty() || !tensor_proto.double_data().empty() || !tensor_proto.int64_data().empty()

报错代码:

# -*- coding: utf-8 -*-
import onnx
import os
import time

import cv2
import numpy as np


net = cv2.dnn.readNetFromONNX(r'onnx_model/0.onnx')
print("net load")

解决方法1:升级opencv版本,升级到4.5.3以上可能就可以。

pip install opencv-python -U

解决方法2:

session = onnxruntime.InferenceSession(onnx_path)

生成代码: 



import argparse
import os

import numpy as np
import onnx
import torch
import torch.nn as nn


class TinyModel(nn.Module):
    def __init__(self, upsample_mode):
        super().__init__()
        self.expander = nn.Conv2d(3, 192, 1, 1)
        upsamples = [nn.Upsample((256, 256), mode='nearest')]
        # upsamples 
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

AI算法网奇

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

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

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

打赏作者

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

抵扣说明:

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

余额充值