ncnn之十：将ncnn的param作为资源文件的调用

原创已于 2024-03-28 13:58:39 修改 · 444 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言

于 2022-06-21 16:24:35 首次发布

C\C++ 同时被 2 个专栏收录

43 篇文章

订阅专栏

ncnn

14 篇文章

订阅专栏

本文介绍了如何在Visual Studio中将NCNN模型的.param文件作为资源添加到项目中，并通过C++代码进行调用。首先，通过右键添加资源并设置资源类型为'param'。然后，在代码中利用GetModuleHandle和FindResource等函数来定位和加载资源，最后通过net.load_param_mem()在ncnn中加载内存中的模型参数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

将ncnn的param作为资源文件的调用

添加资源

Visual Studio里添加资源在所选项目上右键-添加-资源，弹出的窗口里选择-导入，添加我的.param文件，资源类型是"param"，点击确定。
这时项目里会有个 resource.h 文件，里面有类似这样的代码：

#define IDR_CSV1 101
#define IDR_CSV2 102

这里，101 ，102 就是你添加的资源的id。

资源调用

#include "stdafx.h"
#include <iostream>
#include <string>
#include <Windows.h>

using namespace std;

std::string dllPath = L"PCommServerHandler.dll";

LPCWSTR stringToLPCWSTR(std::string orig)
{
	size_t origsize = orig.length() + 1;
	const size_t newsize = 100;
	size_t convertedChars = 0;
	wchar_t* wcstring = (wchar_t*)malloc(sizeof(wchar_t) * (orig.length() - 1));
	mbstowcs_s(&convertedChars, wcstring, origsize, orig.c_str(), _TRUNCATE);

	return wcstring;
}

const char* GetMemParamPtr(const int resourceID)
{
	//定位自定义资源，如果是从本模块定位资源，也可将句柄简单地置为NULL即可
	// HINSTANCE hInst = NULL;
	HINSTANCE hInst = GetModuleHandle(stringToLPCWSTR(dllPath));

	char* paramPtr = NULL;
	HRSRC hrsrc = FindResource(hInst, MAKEINTRESOURCE(resourceID), TEXT("param"));
	if (NULL == hrsrc)
	{
		std::cout << "Find param resource failed!" << std::endl;
		return paramPtr;
	}

	//获取资源的大小
	DWORD dwSize = SizeofResource(hInst, hrsrc);
	if (0 == dwSize)
	{
		std::cout << "Param resource is empty!" << std::endl;
		return paramPtr;
	}

	//加载资源
	HGLOBAL hGlobal = LoadResource(hInst, hrsrc);
	if (NULL == hGlobal)
	{
		std::cout << "Load param resource failed!" << std::endl;
		return paramPtr;
	}

	//锁定资源
	LPVOID pBuffer = LockResource(hGlobal);
	if (NULL == pBuffer)
	{
		std::cout << "Lock param resource failed!" << std::endl;
		return paramPtr;
	}

	paramPtr = (char*)pBuffer;

	return (char*)pBuffer;
}

调用

const char* memParamPtr = GetMemParamPtr(101);

ncnn调用

net.load_param_mem(memParamPtr);

参考资料：
1 Visual Studio添加图片资源到exe
2 C++资源文件初使用[C++菜鸟]