改了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

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐