How to add a new Class (say object refrigerator) to this Yolov8n model
Hello,
Thanks for this sample project. It works great . Now I want to add a new class of object to it say "refrigerator" how can i do that?
Please advise.
Thanks and Regards,
S
Hi @raulp7 Basically this involving training a new YOLO model. This is not possible within Unity as yet, but there are many tutorial online how to do this. You will first need a big dataset of pictures of fridges.
For training some good resources are the website Kaggle, and also a good free tool for training online is called Google Colab. There are some websites where you can train it online and it outputs an ONNX file which you could then use in Unity.
BTW if you would like to vote to have training done in the Unity editor you can vote for it on our roadmap here: https://unity.com/roadmap/unity-ai/sentis
Thanks
@UnityPaul
,
I think adding a way to add more classes and defining a well to do procedure for this will be of great use case for people working in Digital Twin and Industrial Automation using Unity 3D.We are working on many such use use cases and having this feature will be of great use.
It would be good if an official request for this can be added. Please let me know if it is possible and if it can be logged and how?
yes sure I want to vote but got lost in the link you provided. If you can provide the exact place to vote , surely i will.
Regards,
Hi https://unity.com/roadmap/unity-ai/sentis just scroll down to the box that says model training. Click on it then select ones of the buttons at the bottom "nice-to-have" "critical" or "important".
BTW the Sentis team always looks at every post at :unity sentis discussions
And you can also send a direct message here: https://docs.google.com/forms/d/e/1FAIpQLScqFIasPDpRtq-ZjSGJbA0qtO4fCd-aJMWM1d0t3HhEFsBQIw/viewform
Training is something that's going to be very important so we will either have to make it easier for people to train models online or to train them in the editor... or both.
Thanks a lot @UnityPaul !