search

shouxieai / infer

A new tensorrt integrate. Easy to integrate many tasks
MIT License
388 stars 80 forks source link

推理单通道灰度图该怎么修改？ #21

Open fat-921 opened 1 year ago

fat-921 commented 1 year ago

使用yolov8推理单通道的灰度图该怎么修改infer代码？我已经从训练就把模型改成了单通道，但是推理时也想改成单通道灰度图推理。训练灰度图参考了https://github.com/ultralytics/ultralytics/issues/948

UyeYwb commented 11 months ago

请问解决了吗，我这照着修改了，发现了推理无结果