nanodet onnx踩坑记录

原创已于 2022-02-11 23:22:54 修改 · 3.9k 阅读

文章标签：

#深度学习 #机器学习 #计算机视觉

于 2022-02-09 22:34:11 首次发布

深度学习宝典专栏收录该内容

1097 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

作者分享了Nanodet模型在C++和Python中的移植经验，包括C++中使用opencv进行transpose操作（(2,0,1)），以及在Python中的通道变换处理。在C++环境下，作者提到了opencv4.5.4和4.5.5版本在vs2017和2019上的测试成功，并指出mat转vector的修正方法。在Python端，遇到了数组索引错误，提示数组为1维但尝试2维索引。" 104034229,9207076,MySQL5.7.17服务器安装与配置步骤详解,"['数据库', 'MySQL', '服务器安装', '系统配置']

作者分享了挺多的干货：

https://github.com/hpc203

c++版

c++ transpose((2,0,1))

python版

https://github.com/hpc203/nanodet-plus-opencv

c++版

opencv4.5.4 vs2017测试成功

opencv4.5.5 vs2019测试成功

执行：

把模型文件拷贝到

xxx.vcxproj

同级目录即可。

mat转vector时好像不对，我修改了，用convertMat2Vector

template<typename _Tp>
vector<_Tp> convertMat2Vector(const Mat& mat)
{
	return (vector<_Tp>)(mat.reshape(1, 1));//通道数不变，按行转为一行
}

	vector<float> v_mat = convertMat2Vector<float>(dst);


	for (int a = 0; a < 20; a++) {
		cout << v_mat[a] << " ";// ok float

	}
	cout << "\n v_ma