dlunion/tensorRTIntegrate

yolov5 detect gray image

lyxbyr opened this issue · 0 comments

lyxbyr commented

thanks for your works!
i want to detect gray image , i trained yolo with gray image as train_dataset, 现在是灰度图了，转onnx和trt代码与先前的有区别吗？只需要改一下输入的通道数?