jetson_inference是nvidia提供的一个基于tensorrt的推理库,主要用在tx2.xavier这些设备上。不知道用“库”是否准确,可能叫demo更好?可能也不算nvidia正式对外提供的,似乎是nvidia一个员工在维护。在github上可以找得到。
这个项目里的cuda代码主要对图像进行预处理和后处理。
先看这两个:
cudaError_t cudaPreImageNetRGB( float4* input, size_t inputWidth, size_t inputHeight, float* output, size_t outputWidth, size_t outputHeight, cudaStream_t stream );
cudaError_t cudaPreImageNetBGR( float4* input, size_t inputWidth, size_t inputHeight, float* output, size_t outputWidth, size_t outputHeight, cudaStream_t stream );
个人认为主要是做resize的。两个函数的主要区别是颜色通道的顺序不同,因此只看其中一个。
// cudaPreImageNetRGB
cudaError_t cudaPreImageNetRGB( float4* input, size_t inputWidth, size_t inputHeight,
float* output, size_t outputWidth, size_t outputHeight,
cudaStream_t stream )
{