Skip to content

Latest commit

 

History

History
25 lines (14 loc) · 836 Bytes

README.md

File metadata and controls

25 lines (14 loc) · 836 Bytes

FFCNet

FFCNet: Fourier Transform-Based Frequency Learning and Complex Convolutional Network for Colon Disease Classification

Our paper has been accepted by MICCAI 2022.

Prerequisites

Our code is based on python3.6 and pytorch1.1.

Training the networks

python train_test.py

train_dataset-root: Folder to which you downloaded and extracted the training data

val_datapath-root: Folder to which you downloaded and extracted the val data

record_path: The path where the training results are stored

model_path = The path where the model is stored

best_path = The path where the model with the best result on the validation set is stored

First go into the train_test and adapt all the paths to match your file system and the download locations of training and test sets.

Then python train_test.py to train your dataset.