search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.25k stars 195 forks source link

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

Open qqqe2 opened 2 months ago

qqqe2 commented 2 months ago

红外图像或者灰度图像的识别需要更改这里的函数吗？ 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 commented 1 month ago

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