RefineNet语义分割Pytorch模型训练和TensorRT加速

RefineNet模型的训练和ONNX模型导出见我的github：https://github.com/midasklr/RefineNet

TensorRT加速见我的：https://github.com/midasklr/RefineNet_TensorRT

RefineNet模型训练主要参考了 light-weight-refinenet的实现，并做出一些修改方便模型转tensorRT，模型轻巧，加速后可以达到实时分割。

使用的数据集是Helen Dataset，原始的数据集我已经做了一些修改方便分割算法训练，包含了11类面部器官的分割，效果如图：

通过https://github.com/midasklr/RefineNet可以很方便的用于任何分割相关的数据集。

速度：

Model	FPS
Pytorch	5
TensorRT-FP32	27
TensorRT-FP16	33