Dragos-Stan

Hi, I'm having the same error, on both W10 and W11. ultralytics 8.05, training on either object detection or segmentation. Training commands: import ultralytics from ultralytics import YOLO model =...

Thanks :) !

line 903 in utils/general.py prediction = prediction[0][0] # select only inference output