[go: nahoru, domu]

Skip to content

A project to classify parasitized and uninfected malaria pictures using ResNet.

Notifications You must be signed in to change notification settings

pzqkent/Malaria_ResNet

Repository files navigation

Malaria_ResNet

Malaria dataset: https://ceb.nlm.nih.gov/proj/malaria/cell_images.zip

config1.py provides the address of the dataset and the train/dev/test split ratio.

build_dataset1.py will split the dataset into three parts as ratio defined in config1.py

resnet1.py consturct the ResNet.

plot_model.py used to plot the ResNet model

train_model1.py used to train the model as well as plot a graph of several parameters over epochs which will be save as "plot.png" as default.

The model took about 50 minutes to train on AWS EC2 g2.2xlarge instance(NVIDIA GRID K520 GPU) for 20 epochs. Every epoch took about 138s to train and every step of each epoch cost about 223ms.

Got Accuracy = 0.97, Presion = 0.98, Recall = 0.95, F1 = 0.96, F2 = 0.96 on test set(5512 images in total).

Aaron Swartz Aaron Swartz

#credit
The ResNet adapted from book Deep Learning for Computer Vision with Python.
Metrics.py adapted from pape keras-team/keras#5400

About

A project to classify parasitized and uninfected malaria pictures using ResNet.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages