Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

55 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Spinning Up:
1. Create conda environment:

2. Install pytorch in conda environment:
https://pytorch.org/get-started/locally/

3. Install everything in the requirements using pip

4. Download https://github.com/whai362/PVT/releases/download/v2/pvt_v2_b3.pth to ./

5. Create a folder called data_root.
Download and extract https://datasets.simula.no/downloads/kvasir-seg.zip to data_root
Download and extract https://www.dropbox.com/s/p5qe9eotetjnbmq/CVC-ClinicDB.rar?dl=1 to data_root

6. Create a folder called "Trained models" (without the quotations but include the space)
Download and place https://www.dropbox.com/scl/fi/bhfo8ijeg2wyg7p4ehwb7/FCBFormer_both.pt?rlkey=xcjd4mpyfopyziuvut5ry46nv&dl=1 in this folder

Trainings:
There are three jupyter files: teacher_train, student_train_alone, student_train_with_teacher
teacher_train -- used to train the teacher
student_train_alone -- used to train the student alone
student_train_with_teacher -- used to train the student with a teacher

Acknowledgements:
Our work forks the basic training loops used here:
https://github.com/ESandML/FCBFormer

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

2 watching

Forks

Releases

Packages

Used by

Contributors

Languages