P2 output layer

[Argentum471]

🚀 The feature

Model conversion with additional p2 layer.

Motivation, pitch

Hi @zhiqwang,
Im interested to use P2 layer for detecting very small objects in my project. I trained P2 layer from the official YOLOv5's hub. I am using onnx format.

Additional context

• Discussed in P2 detection output #402

^{Originally posted by Argentum471 May 9, 2022}
Im interested to use P2 layer for detecting very small objects in my project. Is it possible to add this in yolort?

[zhiqwang]

Hi, @Argentum471 , Thanks for reporting this ticket, and more specifically, do you want to use it on TensorRT or ONNX Runtime?

[Argentum471]

ONNX Runtime