改了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
文章讨论了在使用Yolov8进行训练时,更换predictor中的Warmup图像从`assert/bus.jpg`到符合数据集的图片导致的问题。具体提到需要修改`ultralytics/data/loaders.py`中`LoadImages`类的`__next__`函数,确保图片通道匹配。解决方法涉及`cv2.imread`和`getMulti2Array_255_apart`函数的调用与路径处理。
138

被折叠的 条评论
为什么被折叠?



