[D] Extracting the class labels and bounding boxes for objects, from a YOLO7 model after converting to an ONNX model

This page summarizes the projects mentioned and recommended in the original post on /r/MachineLearning

Scout Monitoring - Free Django app performance insights with Scout Monitoring
Get Scout setup in minutes, and let us sweat the small stuff. A couple lines in settings.py is all you need to start monitoring your apps. Sign up for our free tier today.
www.scoutapm.com
featured
InfluxDB - Power Real-Time Data Analytics at Scale
Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
www.influxdata.com
featured
  • yolov7

    Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

    (Please note, this is a re-post of my original question here, I think this subreddit might be more appropriate for asking this question)At work, we use Unity, we have a project that needs object detection and classification. We decided to use this YOLO7 model (for non-technical reasons, It had to be the exact same model as the company does have pre-trained weights for this exact model). However, Unity only supports ONNX so I exported the model as an ONNX model, using the code provided in the repo:

  • Scout Monitoring

    Free Django app performance insights with Scout Monitoring. Get Scout setup in minutes, and let us sweat the small stuff. A couple lines in settings.py is all you need to start monitoring your apps. Sign up for our free tier today.

    Scout Monitoring logo
  • barracuda-release

    I removed --end2end from the export flags, which removed the warning, and for the Resize warning I followed the advice on a similar issue

  • ONNX-YOLOv7-Object-Detection

    Python scripts performing object detection using the YOLOv7 model in ONNX.

    Finally, I tried to look if someone has done similar work for the ONNX model and I found this repo which links the same repo I am trying to use, and I believe this function is doing exactly what I want to do, but I could not understand what it is doing (I don't understand how it knows exactly where the number of detections is, and where the bounding boxes are and the class labels, etc.) furthermore, I am not sure if removing end2end and the changing the version from 12 to 9 has any effect on the output shape or it has to do with the internal layers.

  • yolov5

    YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

    Those dimensions suggest you need to apply (i.e. roll your own) non-max suppresion to the outputs: relevant link

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts

  • A video based Latin dictionary: get what you see in Latin (beta) - What do you think?

    1 project | /r/latin | 23 Apr 2023
  • Coding Question Help

    1 project | /r/learnprogramming | 4 Mar 2023
  • DL for the Web: Repository of Models

    2 projects | /r/deeplearning | 27 Feb 2023
  • [OC] Football Player 3D Pose Estimation using YOLOv7 and Matplotlib

    1 project | /r/dataisbeautiful | 19 Feb 2023
  • Finding a good Tiny Yolo to train in Python

    1 project | /r/deeplearning | 17 Feb 2023

Did you konow that Python is
the 1st most popular programming language
based on number of metions?