thb1314/tensorrt-onnx-fasterrcnn-fpn-roialign

How to use my own model?

Closed this issue · 3 comments

是否可以使用自己在pytorch faster-rcnn上训练得出的权重模型转onnx呢?具体需要怎么操作呢?谢谢!

@Nuralill 可以采用torchvision上的faster-rcnn模型,然后训练脚本需要自己写一下。

那就是只支持resnet50吧?

@Nuralill 需要自行适配,其他的也可以支持的