duan8/swin-transformer/semantic-segmentation

Swin Transform - Semantic Segmentation

The Pytorch implementation is microsoft/Swin-Transformer.

Only support Swin-T, welcome the PR for other backbones.

Authors

How to Run

1. generate .wts from pytorch with .pt, or download .wts from model zoo

git clone https://github.com/microsoft/Swin-Transformer.git
git clone https://github.com/wang-xinyu/tensorrtx.git

python gen_wts.py Swin-Transform.pt
// a file 'Swin-Transform.wts' will be generated.

2. build tensorrtx/swin-transform and run

cd {tensorrtx}/swin-transform/semantic-segmentation/
mkdir build
cd build
cp {microsoft}/Swin-Transformer/Swin-Transform.wts {tensorrtx}/swin-transformer/semantic-segmentation/build
cmake ..
make
sudo ./swintransformer -s [.wts] [.engine]   // serialize model to plan file
sudo ./swintransformer -d [.engine] [image folder]  // deserialize and run inference, the images in [image folder] will be processed.

More Information

See the readme in home page.