How to get correct yolo11n web model ?

[Paliking]

Hi,

thank you for your great project. I am facing an issue when I use my own yolo11n model. How did you create your yolo11n_web_model in this project?

When I create webmodel this way and place it to your project, then I get below error

my command to crete web model
yolo export model=yolo11n.pt format=tfjs

error after this javascript line from file App.jsx: const warmupResults = yolo.execute(dummyInput);
Error: Error in conv2d: depth of input (128) must match input depth for filter 1.

All is working fine if I use e.g. yolov8n created by this command:
yolo export model=yolov8n.pt format=tfjs

I tried combinatiosn of multiple versions of tensorflow and tensorflowjs, but nothing helped. Does it work for you guys?

thank you

[sitammeur]

Yes, I can help you. I used a notebook to convert the standard YOLO model with PyTorch weights to the TensorFlow.js model format.
Colab Link: https://colab.research.google.com/drive/1hrfUyK5N-1ctXuULL3WSjSz4UmAlL1gW?usp=sharing

[Paliking]

Hi sitamgithub-MSIT,

thank you, but it didnt help. I still get the same error. Could it be due to changes in other library versions in Colab? If you run the Colab notebook you shared right now, would you get a working yolo11n model for this project?

thank you

[sitammeur]

I think so. You should get your model weight and directly run my colab for export purposes.

[sitammeur]

By the way, are you using the standard Yolo11 model for the COCO dataset, or is it your fine-tuned model for the custom dataset?

[Paliking]

I executed the colab notebook you shared, downloaded yolo11n web model and inserted to your project. When I load it, then I get the error I mentioned. Are you sure it still works with the current libraries versions in the current colab notebook?

thank you

[sitammeur]

Ok, I get the same problem. I believe it is a model update, or there is a change, most likely when exporting to the TensorFlow.js format. As of now, I will say to use my already having model weights or move to YOLO12 if possible.

[Paliking]

thank you for the confirmation. I face the same issue with yolo12n too. It is working with yolov8n only for me.

[sitammeur]

@Paliking Just an update: I have made code changes, and the model weights are updated in the same notebook. The app works now, so your problem may be solved.