TESN: Transformers Enhanced Segmentation Network
We proposed Transformers Enhanced Segmentation Network (TESN) which made the following improvements to the original Mask R-CNN to improve the accuracy of instance segmentation: a) TESN adopt multi-stage architecture to provide high-quality target detection. b) TESN Introduce Transformer into the mask head to provide high-quality segmentation.
Prepared to train
python 3.7.10 cuda 10.1 torch 1.5.0 mmcv-full 1.3.8 mmdet 2.18.0
1)Place fcn_mask_transformer_head.py and transformer.py under ../mmdetection/mmdet/models/roi_heads/mask_heads 2)Place TESN_COCO.py under ../mmdetection 3)add ‘from .fcn_mask_transformer_head import FCNMaskTransformerHead’ into ../mmdetection/mmdet/models/roi_heads/mask_heads/init.py
Run ‘python tools/train.py TESN_COCO.py’
Run ‘python tools/test.py TESN_COCO.py’