search

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile
464 stars 114 forks source link

yolov5 detect gray image #48

Closed lyxbyr closed 3 years ago

lyxbyr commented 3 years ago

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