yolov7
yolov7 copied to clipboard
Add grid concat and fuse such operators
Now we can export concat all anchors outputs in flag --grid
!
If we use --grid
only, we will export an onnx with one concat output.
If we use --grid
and --end2end
, we will export an onnx with nms.
And now the anchor generate process is simplied as the picture shown!
It will speed up in onnxruntime and tensorrt backends!
Also do you have any test, does it work faster and still correctly?
@AlexeyAB I have test in batch=32 in cpu onnxruntime. Two results are the same. This pr speed up 2-3%. If we use huge batchsize, maybe speed up more!