search

cyrusbehr / tensorrt-cpp-api

TensorRT C++ API Tutorial
MIT License
596 stars 74 forks source link

Original function: #60

Closed qq978358810 closed 4 months ago

qq978358810 commented 4 months ago 
static cv::cuda::GpuMat blobFromGpuMats(const std::vector<cv::cuda::GpuMat> &batchInput, const std::array<float, 3> &subVals,
                                        const std::array<float, 3> &divVals, bool normalize);

Modified function:

static cv::cuda::GpuMat blobFromGpuMats(const std::vector<cv::cuda::GpuMat> &batchInput, const std::array<float, 3> &subVals,
                                        const std::array<float, 3> &divVals, bool normalize, bool swapRB = false);

In this modification, an additional parameter swapRB of type bool with a default value of false is added to the function signature. This parameter is used to specify whether the Red and Blue channels should be swapped during processing.

cyrusbehr commented 4 months ago

@all-contributors please add @qq978358810 for code

allcontributors[bot] commented 4 months ago

@cyrusbehr

I've put up a pull request to add @qq978358810! :tada: