/SwinDetr

Integration of Swin Transformer to DETR for Robust Object Detection (DEMO)

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

SwinDetr

Swin Transformer Model Architecture

DETR Model Architecture

This repo hosts the integration of vision transformer, namely Swin Transformer, to the transformer based Object Detection model, namely DETR for robust Object Detection. Swin transformer implementation is based on here and DETR is based on DETR. For the sake of simplicity, demo version of DETR is used. Please contact me for further questions.