P2 detection output

[Argentum471]

Hello @zhiqwang,
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 ,

yolort does not currently support custom P2 structures, but it would not be very difficult to do so. Did you get your P2 layer from the official YOLOv5's hub?

I think it will be better to discuss this ticket in the issue to make it easier to track it, would you like to resubmit a new ticket labeled enhancement at here, and it would be better to mention your subsequent usage as well. Because the changes we make for different backends compatibility for the LibTorch, ONNX Runtime and TensorRT will require more work, but it would be much easier to support just one of them.