Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Change the order of labels #3841

Closed
tetelevm opened this issue Jun 30, 2021 · 1 comment · Fixed by #3843
Closed

Change the order of labels #3841

tetelevm opened this issue Jun 30, 2021 · 1 comment · Fixed by #3843
Labels
question Further information is requested

Comments

@tetelevm
Copy link

(It's funny that the question has a red emoji and a purple tag, and the bug has a purple emoji and a red tag)

To display the labels on a photo I use the code

    def get_predict(self): 
        predict = self.model(image)
        return predict.render()[0]

And there is a problem, which can be seen in the image

img

As I understand it, rendering goes on probability reduction, and because of this objects with lower probability can block objects with higher probability. How to make the drawing of labels in reverse, from less to more?

@tetelevm tetelevm added the question Further information is requested label Jun 30, 2021
@glenn-jocher glenn-jocher linked a pull request Jun 30, 2021 that will close this issue
@glenn-jocher
Copy link
Member

glenn-jocher commented Jun 30, 2021

@tetelevm good news 😃! Your original issue may now be fixed ✅ in PR #3843. To receive this update:

  • Gitgit pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
  • PyTorch Hub – Force-reload with model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
  • Notebooks – View updated notebooks Open In Colab Open In Kaggle
  • Dockersudo docker pull ultralytics/yolov5:latest to update your image Docker Pulls

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants