Skip to content
/ STFT Public
forked from lingyunwu14/STFT

Spatial-Temporal Feature Transformation for Video Object Detection, MICCAI2021

License

Notifications You must be signed in to change notification settings

Lechtr/STFT

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

STFT: Spatial-Temporal Feature Transformation

forked for the Master Thesis "Diagnosis of Polyps with focus on video analysis" https://github.com/Lechtr/STFT

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

  • [2021/11/12] For the implementation on the ImageNet VID dataset, please refer to here.
  • [2021/09/21] Implementation for other video-based methods on Endoscopic Video Datasets released.
  • [2021/09/21] Release training/testing scripts and the pretrained model for STFT.
  • [2021/06/29] Create repository.

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.

About

Spatial-Temporal Feature Transformation for Video Object Detection, MICCAI2021

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 65.3%
  • Python 28.7%
  • Cuda 4.8%
  • Other 1.2%