如何将训练好的swin.pth文件转成tensortRT并进行推理，目标检测有这个脚本，分割没有该脚本嘛？

SwinTransformer / Swin-Transformer-Semantic-Segmentation

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.

https://arxiv.org/abs/2103.14030

Apache License 2.0

1.18k stars 222 forks source link

Open jeromezyb opened 2 years ago