Skip to content

Bugi-Sulistiyo/Comparing-EfficientNet-and-MobileNet-on-Glaucoma-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Glaucoma Detection

This project is my research to complete my bachelor degree at Mulawarman University. The title of my research is A Comparison of MobileNet and EfficientNet Architechture on Glaucoma Classification Based on Fundus Image. The result could be seen on dashboard at bit.ly/dashboard_skripsi_bugi.

Project Steps

1. Data Collecting

The dataset used for building the model is fundus image. There are 4 fundus image dataset used. This 4 datasets choosen because it is clinically diagnosed by an ophthalmologist.

  • G1020 - dataset - paper
    This dataset is collected at a private clinical practice in Kaiserslautern, Germany between year 2005 and 2017. It consisted of 1020 images from 432 patients. The images are labeled as glaucoma (296 images) and healty (724 images).

  • RIM-ONE - dataset - paper
    This dataset were taken at Universitario de Canarian Hospital and Universitario Miguel Hospital in Madrid, Spanish. It was taken in year 2011(v1), 2014(v2), 2015(v3). The one used for this project is RIM-ONE DL that make specificcally for computeri vision dataset. It consist of 485 fundus images labeled as glaucoma (172) and normal (313).

  • REFUGE - dataset - paper
    This dataset is a competition dataset held by MICCAI on 2018. It consist of 1200 fundus images labeled as glaucoma (1080) and normal (120).

  • PAPILA - dataset - paper
    This dataset were taken at General Universitario Reino Sofia Hospital in Murcia, Spain between 2018-2020. It consist of 488 fundus images labeled as glaucoma, suspicious, and normal.

2. Preparation and EDA

Before all the dataset being augmented in preprocessing step, it need to be restructurize. The dataset should be restructurize because some of it have a different structure file and some of it does not devided by folder for each label. In order to do the restructurization of the file structure, the dataset must be explored and the metadata is understood. Also, data splitting process is being done in this step.

3. Preprocessing

The preprocessing methoed used is augmentation. The augmentation used are horizontal & vertical flipping, brightness, and CLAHE.

4. Training Model

There are 6 model trained using 4 dataset. I used 3 newest model from EfficientNet and MobileNet architecture each. all model is trained on every dataset and scenario defined.

5. Testing Model

After training model is completed, the model is immidiatly saved in .h5 format. After that, the testing script load each model and test it on every scenario.

6. Dashboard Development

To make the result easier to compared, I make a dashboard to show the final result.

File Structure

  • data\ --> (hidden) store train result, model, and all important information
  • dataset\ --> (hidden) hide because have a large size. store source dataset, prepared dataset, and augmented dataset
  • lampiran\ --> store all file from data\ that will be attached on skripsi
  • main\scripts\ --> store all the script used in this project

Replication Procedures

  1. Download all the dataset mentioned above and stored it in dataset directory.
  2. Prepare global variable stored in .env file. All varieble used could be seen on var_guide.txt
  3. Run the existing script based on the numbering of existing directories and files