- 博客(6)
- 资源 (4)
- 收藏
- 关注
RSS订阅原创 onnx转tensorrt报错Assertion Error in getSupportedFormats: 0 的解决方法
为了让分割模型的结果保留每个像素点的置信度，修改onnx模型（删除argmax层），重新转trt时发生报错这类报错在网上搜不到特别有用的情况，于是用onnxruntime跑一遍产生问题的onnx模型，报错如下可以看出是output层和上一层之间的数据类型不匹配继续查找资料发现有大神已经做过此类修改方法https://blog.youkuaiyun.com/weixin_43945848/article/details/122486725把输出层的elem_type修改为1后，问题解决。output能输出包含每
2022-05-23 10:03:10
505
原创 QT调试报错 Cannot mix incompatible Qt library (version ...) with this library (version ...)的问题解决
今天调试一份qt代码遇到了qtlib不匹配的问题网上查的资料包括创建文件夹mv qt的库，qt版本冲突等，均未生效。通过单步代码调试发现主要的模型推理代码都正常报错信息Cannot mix incompatible Qt library (version …) with this library (version …)位于cv::imshow的地方，我把cv::imshow相关代码注释后，程序能正常运行完。这次问题估计是cv::imshow的弹窗和qtcreator不太匹配，值得注意的是其他cv
2022-05-19 11:00:49
3242
原创 pytorch上实现简单的定点网络
最近开始学pytorch相关的网络模型训练，作为基础练习训练一个关键点检测模型。入坑看了这个项目 说明的还是很详细的https://github.com/tensor-yu/PyTorch_Tutorial里面的loss函数介绍以处理分类模型为主，而数据处理和读取，pytorch训练流程都讲的很全。因此我这次练点检测的模型只要稍微改一下数据读取和loss的计算即可。我目前生成的数据保存在txt里，一行包含的信息为“图像路径+4个关键点的八个坐标”，因此在自定义数据模块只做了如下处理：（一般放在uti
2022-03-16 17:29:33
2141
原创 用python实现caffe上ssd的检测结果输出
如果deploy.prototxt没有写错的话，可以跑出相应的检测结果（坐标和分类）import numpy as np import sys,os import cv2caffe_root = '/work/caffessd' #caffe目录sys.path.insert(0, caffe_root + 'python') import caffe import ...
2019-08-06 09:32:36
905
原创 海思Hi3519A NNIE开发中mobilenet-ssd模型转换过程中遇到的坑
我之前用的mobilenet-ssd模型中深度可分离卷积（dw）的prototxt如下通过group来实现（代码中就是输入的8个featuremap经过分8组，每组各和一个卷积核做卷积运算，最后运算输出还是8个featuremap）layer { name: "conv1/dw" type: "Convolution" bottom: "conv0" top: "conv1/d...
2019-04-15 15:25:28
5930
9
lmdb文件随机删减与乱序合并.rar
2020-01-17
通过layer名称对caffemodel进行剪枝
2019-12-10
计算网络生成结果iou准确率.zip
2019-12-06
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人