颜色空间转换:基于CUDA的BGR转HLS

基于paddledetection或者paddlex如何训练模型就不再这说了,具体参考官方文档,接下来主要是模型如何本地部署并用C#进行调用。我用的官方给的从PaddleDetection中导出的YOLOv3模型,大家自行下载具体链接:

C++

__global__ void BGR2HLS(uchar3 *img, uchar3 *imgout, int ImgWidth, int imgHeigh)
{
    int ImgNumColonne = blockIdx.x  * blockDim.x + threadIdx.x; //idx  //=====col====
    int ImgNumLigne = blockIdx.y  * blockDim.y + threadIdx.y;  //idy  //=====row====
    int Index = (ImgNumLigne * ImgWidth + ImgNumColonne );


    if ((ImgNumColonne < ImgWidth ) && (ImgNumLigne < imgHeigh))
    {
        /* Kernel Code Here */
        double blue = (double)img[Index].x / 255;
        double green = (double)img[Index].y / 255;
        double red = (double)img[Index].z / 255;


        double cMax = maxVal(blue, green, red);


        double cMin = minVal(blue, green, red);


      

猜你喜欢

转载自blog.csdn.net/zhangdaoliang1/article/details/124862047