Home

Awesome

STFT: Spatial-Temporal Feature Transformation

License

By Lingyun Wu, Zhiqiang Hu, Yuanfeng Ji, Ping Luo, Shaoting Zhang.

This repo is an PyTorch implementation of "Multi-frame Collaboration for Effective Endoscopic Video Polyp Detection via Spatial-Temporal Feature Transformation", accepted by MICCAI 2021.

Framework Overview

This repository contains the implementation of our approach STFT and several other video object detection algorithms like FGFA, RDN, and MEGA based on mega.pytorch, as well as training and testing scripts to reproduce the results on Endoscopic Video Datasets reported in our paper.

News

Model Zoo

Supported backbones:

Supported image-based methods:

Supported video-based methods:

Installation

Please follow INSTALL.md for installation instructions.

Usage

Please follow GetStarted.md for usage instructions.

Citing STFT

Any new methods are welcomed. We also hope this repository would help further research in the field of video object detection and beyond. Please cite our paper in your publications if it helps your research:

@inproceedings{wu2021multi,
  title={Multi-frame collaboration for effective endoscopic video polyp detection via spatial-temporal feature transformation},
  author={Wu, Lingyun and Hu, Zhiqiang and Ji, Yuanfeng and Luo, Ping and Zhang, Shaoting},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={302--312},
  year={2021},
  organization={Springer}
}

Contributing to the project

Any pull requests or issues are welcomed.