search

Rapternmn / PyTorch-Onnx-Tensorrt

A set of tool which would make your life easier with Tensorrt and Onnxruntime. This Repo is designed for YoloV3
80 stars 18 forks source link

maxpool node #5

Open gillmac13 opened 4 years ago

gillmac13 commented 4 years ago

@Rapternmn I wonder how difficult it would be to integrate a maxpool node in the ONNX conversion. That would enable converting yolov3-spp and yolov3-tiny models which are also very popular.

gillmac13 commented 4 years ago

@Rapternmn OK, don't worry.. The original "yolov3_to_onnx.py" by Nvidia provides all the necessary nodes, including maxpool. I'll give it a try and make a yolov3-spp model in ONNX and RT formats. Thanks for your work!