How do I change to use yolov11's model?

[Jeremy-zhangyichen]

How do I change to use yolov11's model?

[JiayuanWang-JW]

Unfortunately, this repo can not follow the Ultralytics update. If you want to use yolov11's model, you have to add the new modules and build the model yaml file following YOLOv11.

[gp01pw]

@JiayuanWang-JW

Thank you so much for nice work...
I tried building this using YOLOV11. I have attached corresponding yaml file. I am getting following error. Can you help me debug this error?

I am getting error at loss function. This line

loss[0] = self.TL(preds, binary_mask, 0.7).

But I believe it's issue of model's output at engine-> trainer
if self.args.task == "multi":
preds = self.model(batch[0]['img'])

The pred size I get is (batch, 2, 40, 40) which should be (batch, 2, 640, 640)

log.txt

yolov11-bdd-v4-one-dropout-individual-l.md