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