jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.75k stars 547 forks source link

Allow arbitrary number of anchors (masks) in YOLO layer #540

Closed ChristopheKar closed 2 years ago

ChristopheKar commented 2 years ago

This allows for models like YOLOv4-P5 with 4 anchors in each YOLO layer to be converted to ONNX and then to TensorRT, by simply parsing the configuration file. Tested for YOLOv4-P5 with dimensions 896 (original config dimensions) and 448.