Skip to content

tobiascz/TeCNO

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Table of Contents

logo tecno

  1. About
  2. Getting started
    1. Prerequisites
    2. Usage
  3. Reference
  4. Contact

About

TeCNO performs hierarchical prediction refinement with causal, dilated convolutions for surgical phase recognition and outperforms various state-of-the-art LSTM approaches!

Link to paper: TeCNO Paper

logo tecno

Getting started

Follow these steps to get the code running on your local machine!

Prerequisites

pip install -r requirements.txt

Usage

We are using the publicly available Cholec80 dataset. For training we split the videos into individual frames.

Stage 1 - Train Feature Extractor

Run:

python train.py -c modules/cnn/config/config_feature_extract.yml

This will train your feature extractor and in the Test Step it will extract for each Video the features of all images and save it as .pkl

Stage 2 - Train Temporal Convolutional Network

python train.py -c modules/mstcn/config/config_tcn.yml

Reference

@inproceedings{czempiel2020,
 author    = {Tobias Czempiel and
               Magdalini Paschali and
               Matthias Keicher and
               Walter Simson and
               Hubertus Feussner and
               Seong Tae Kim and
               Nassir Navab},
 title     = {TeCNO: Surgical Phase Recognition with Multi-Stage Temporal Convolutional
               Networks},
  booktitle = {Medical Image Computing and Computer Assisted Intervention - {MICCAI}
               2020 - 23nd International Conference, Shenzhen, China, October 4-8,
               2020, Proceedings, Part {III}},
  series    = {Lecture Notes in Computer Science},
  volume    = {12263},
  pages     = {343--352},
  publisher = {Springer},
  year      = {2020},
}

Contact

For any problems and question please open an Issue

Follow me on alt text

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages