Retinal Optical Coherence Tomography Classification

Pytorch implementation for retinal OCT images of Kaggle

Model

Results

Prerequisites

• Pytorch
• Retinal OCT Image Dataset

Getting Started

Installation

• Install PyTorch and dependencies from http://pytorch.org
• Install Torch vision from the source.

git clone https://github.com/pytorch/vision
cd vision
python setup.py install

Retinal OCT Image Dataset

• To train the model, modify the dataset path in train.py

data_dir = 'dataset path'

Train

• Train a model:

python train.py 

Test

• Test the model:

python test.py