rknpu2里的yolov5demo，怎么输入灰度图

[LiLiLiam]

src = wrapbuffer_virtualaddr((void *)img.data, img_width, img_height, RK_FORMAT_RGB_888); dst = wrapbuffer_virtualaddr((void *)resize_buf, width, height, RK_FORMAT_RGB_888);

demo里的默认的输入图像时三通道的，我现在需要输入灰度图应该怎么改？

[youxin1996]

直接把灰度转RGB是最简单的办法，不然你就得改模型结构，并且只能减少第一层的参数，综合性能提升不到10%