YOLOV5 如何输入灰度图进行检测

Question

YOLOV5 如何输入灰度图进行检测? 是否需要修改模型输入？我在正常模型内输入灰度图检测会比输入三通道图像更加慢，理论上三通道变单通道不是应该更快吗？

Answer 1

默认的模型就是三通道，不管你怎么输入，不改模型的话，你输入单通道的他需要转成三通道之后再进行。
而单通道的话，需要你自己去修改网络结构和模型，但是实际上改成单通道参数量并没有下降多少（原始模型的95-98%），速度提升不了多少的。