opencv中的Mat传入到TensorRT中

核函数线程索引与图像对应关系

        在opencv中进行图像像素索引,首先在width上索引,再在height上索引。两个循环的外层是height,内层是width。在核函数中的更新速度x>y>z,当只使用二维网格时,x对应图像width,y对应图像height。

对图像进行归一化

        在resnet中将像素从0-255映射到0-1之间,通过减去均值除以方差。

float __device__ normalized(float src, float mean, float std)
{
float res = (src / 255.0f - mean) / std;
return res;
}

HWC转化为CHW

由于通过opencv读取到的图像为BGR格式,通过指针进行逐像素索引的顺序是BGRBGRBGR,在tensorRT中传入的像素顺序由神经网络模型推理的通道顺序确定的,在resnet中tensor为[C,H,W]的顺序传入的。

void __global__ preprocess(uchar * src, float* res, const int width, const int height)
{
int x = threadIdx.x + blockIdx.x * blockDim.x;
int y = threadIdx.y + blockIdx.y * blockDim.y;
const int size = width * height;
const float mean[3] = {0.485f, 0.456f, 0.406f};
const float std[3] = {0.229f, 0.224f, 0
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值