红外图像或者灰度图像的识别

[qqqe2]

红外图像或者灰度图像的识别需要更改这里的函数吗？ resizeDevice(m_param.batch_size, m_input_src_device, m_param.src_w, m_param.src_h, m_input_resize_device, m_param.dst_w, m_param.dst_h, 114, m_dst2src);

bgr2rgbDevice(m_param.batch_size, m_input_resize_device, m_param.dst_w, m_param.dst_h,
    m_input_rgb_device, m_param.dst_w, m_param.dst_h);
normDevice(m_param.batch_size, m_input_rgb_device, m_param.dst_w, m_param.dst_h,
    m_input_norm_device, m_param.dst_w, m_param.dst_h, m_param);
hwc2chwDevice(m_param.batch_size, m_input_norm_device, m_param.dst_w, m_param.dst_h,
    m_input_hwc_device, m_param.dst_w, m_param.dst_h);

[FeiYull]

@qqqe2 如果你的图像是类似灰度图的单通道图像，此时是不用的