macreho Goto Github PK

 Vignav Ramesh

arXiv / Full Paper (PDF) / Papers With Code / Website (Malaria Diagnosis Algorithm, 3D Cell Rendering)

Our models were trained on a single GPU (Tesla P4 GPU provided by Google Colab, 16 GB memory), but the code can be run as is on a CPU (with increased training times). The code is implemented using Python 3, Keras, and TensorFlow v2. To install all required dependencies, run the following:

pip3 install -r requirements.txt

All data is stored in this repository and can be accessed here.

Use the following Colab file to train the model:

@misc{ramesh2021endtoend,
  title={End-to-end Malaria Diagnosis and 3D Cell Rendering with Deep Learning},
  author={Vignav Ramesh},
  year={2021},
  eprint={2108.04220},
  archivePrefix={arXiv},
  primaryClass={cs.CV}
}