yolov8 多通道predict

你好123456789012

已于 2024-04-18 17:54:00 修改

阅读量200

点赞数 1

文章标签： YOLO

于 2023-12-03 13:52:39 首次发布

本文链接：https://blog.csdn.net/huachenyuyu1/article/details/134764237

版权

改了predictor 却影响了正常train

这是因为模型推理时的warmup步骤使用了assert/bus.jpg

如果替换成符合你数据集的其他图片。和yolov8.pt的三通道对不上

因此修改ultralytics/data/loaders.py

LoadImages类

__next__函数中

else:
    # Read image
    self.count += 1
    im0 = cv2.imread(path)  # BGR
    if im0 is None:
        raise FileNotFoundError(f'Image Not Found {path}')
    # sxxpred
    mx = getMulti2Array_255_apart(path)  # 得到array
    if len(mx)>0:
        im0 = np.concatenate((im0, mx), axis=2)

    s = f'image {self.count}/{self.nf} {path}: '

return [path], [im0], self.cap, s