Yolov5 predicting single class

[Una865]

I am training the YOLOv5 model on predicting Sign Language. However, the I train the model it outputs many classes for one single image(one sign shown) and I want to have prediction based on the most confident class and have only that class shown. How can I do that?

[glenn-jocher]

@Una865 you can modify max detections parameter down to 1

[Una865]

How can I do that in Google Colab, I cloned the GitHub yolov5

[glenn-jocher]

See https://colab.research.google.com/github/ultralytics/yolov5/blob/master/tutorial.ipynb?hl=en

[Una865]

!python detect.py --weights runs/train/exp2/weights/best.pt --img 416 --conf 0.1 --max_det 1 --source '/content/American-Sign-Language-Letters-1/test/images'
This does not quite works

[glenn-jocher]

@Una865 👋 hi, thanks for letting us know about this possible problem with YOLOv5 🚀. We've created a few short guidelines below to help users provide what we need in order to start investigating a possible problem.

How to create a Minimal, Reproducible Example

When asking a question, people will be better able to provide help if you provide code that they can easily understand and use to reproduce the problem. This is referred to by community members as creating a minimum reproducible example. Your code that reproduces the problem should be:

• ✅ Minimal – Use as little code as possible to produce the problem
• ✅ Complete – Provide all parts someone else needs to reproduce the problem
• ✅ Reproducible – Test the code you're about to provide to make sure it reproduces the problem

For Ultralytics to provide assistance your code should also be:

• ✅ Current – Verify that your code is up-to-date with GitHub master, and if necessary git pull or git clone a new copy to ensure your problem has not already been solved in master.
• ✅ Unmodified – Your problem must be reproducible using official YOLOv5 code without changes. Ultralytics does not provide support for custom code ⚠️.

If you believe your problem meets all the above criteria, please close this issue and raise a new one using the 🐛 Bug Report template with a minimum reproducible example to help us better understand and diagnose your problem.

Thank you! 😃