/Malaria_detect

Primary LanguageJupyter Notebook

Malaria_detect

This repo has two Deep Learning models to classify malaria parasite infected cells.

  • Malaria_detect.ipynb: Contains a custom architecture model.
  • Malaria_Detect.ipynb: Contains a Resnet34 rchtecture and traied the model using transfer learning.

Dataset: https://www.kaggle.com/iarunava/cell-images-for-detecting-malaria