Merge classification and detection Model onnx

[2022Mahdi]

Search before asking

• I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

im trying to Merge two models first one is a detection model and i would like to feed detected object to a classifier model both model traind by yolov5 and converted to onnx , i need an onnx model that get an image and use both models to detect and classify object

note:i know i can directly classfiy and detecte object using detection model but i prefer to have more accurate model

Additional

No response

[glenn-jocher]

@2022Mahdi to accomplish this, you can load both ONNX models within a single script and run the detection model followed by the classification model on the detected objects. After obtaining the detection results, you will need to extract the bounding boxes of the detected objects and crop them out of the original image to feed to the classification model. You can then merge the results of the two models to obtain the final predictions. You can use the ONNX runtime to execute the models within your script. For more information on using ONNX for object detection and classification, please refer to the Ultralytics YOLOv5 documentation available at https://docs.ultralytics.com/yolov5/.

[github-actions]

👋 Hello there! We wanted to give you a friendly reminder that this issue has not had any recent activity and may be closed soon, but don't worry - you can always reopen it if needed. If you still have any questions or concerns, please feel free to let us know how we can help.

For additional resources and information, please see the links below:

• Docs: https://docs.ultralytics.com
• HUB: https://hub.ultralytics.com
• Community: https://community.ultralytics.com

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLO 🚀 and Vision AI ⭐