海滨的Blog

Good Luck To You!

FasterCNNN+ROIAlign+FPN在TensorRT上的部署的解决方案。

FasterRCNN作为经典的two-staged detector,至今为止其精度配合FPN+ROIAlign依然SOTA。

然而FasterRCNN在TensorRT的部署却没有one-stage的检测器那么容易。因为TensorRT输入输出要求shape是确定值,就算新的版本支持动态batch_size,但是在一开始也要确定最大输入的batch_size,中间的算子是不允许运行时更改维度的。

<< 1 >>

Powered By Z-BlogPHP 1.7.1

Copyright hbblog.cn Rights Reserved.