- Introduction
- NVIDIA Jetson Nano
- TFRT
- ONNX
- TensorRT
- Motivation
- DISCLAIMER
- Acute Lymphoblastic Leukemia
- ALL IDB
- Getting Started
- Quick Start
- Contributing
- Versioning
- License
- Bugs/Issues
The Acute Lymphoblastic Leukemia Jetson Nano Classifier is a Convolutional Neural Network (CNN) developed using Intel® oneAPI AI Analytics Toolkit and Intel® Optimization for Tensorflow* to accelerate the training process, and TensorRT for high performance inference on the NVIDIA® Jetson Nano™.
In this project you will do the following.
- Train a custom CNN for Acute Lymphoblastic Leukemia on your development machine.
- Convert the Tensorflow SavedModel to TFRT format
- Convert the Tensorflow SavedModel to ONNX format
- Convert the Tensorflow ONNX model to TensorRT format
- Test the CNN on your development machine.
- Download or create a custom docker container for your NVIDIA Jetson Nano.
- Run the CNN on your NVIDIA Jetson Nano using Tensorflow.
- Run the CNN on your NVIDIA Jetson Nano using TFRT.
- Run the CNN on your NVIDIA Jetson Nano using TensorRT.
The NVIDIA® Jetson Nano™ is a small, powerful computer that is capable of running multiple neural networks in parallel for applications like image classification, object detection, segmentation, and speech processing.
"TFRT is a new TensorFlow runtime. It aims to provide a unified, extensible infrastructure layer with best-in-class performance across a wide variety of domain specific hardware. It provides efficient use of multithreaded host CPUs, supports fully asynchronous programming models, and focuses on low-level efficiency."
"ONNX is an open format built to represent machine learning models. ONNX defines a common set of operators - the building blocks of machine learning and deep learning models - and a common file format to enable AI developers to use models with a variety of frameworks, tools, runtimes, and compilers."
TensorRT
"NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference. It includes a deep learning inference optimizer and runtime that delivers low latency and high throughput for deep learning inference applications."
The motivation behind this project was to test the capabilities of the NVIDIA® Jetson Nano™ for the classification of Acute Lymphoblastic Leukemia on the edge. This project is also a submission for the NVIDIA Jetson AI Specialist Certification.
This project should be used for research purposes only. The purpose of the project is to show the potential of Artificial Intelligence for medical support systems such as diagnostic systems.
Although the model is accurate and shows good results both on paper and in real world testing, it is trained on a small amount of data and needs to be trained on larger datasets to really evaluate it's accuracy.
Developers that have contributed to this repository have experience in using Artificial Intelligence for detecting certain types of cancer. They are not doctors, medical or cancer experts.
Acute lymphoblastic leukemia (ALL), also known as Acute Lymphocytic Leukemia, is a cancer that affects the lymphoid blood cell lineage. It is the most common leukemia in children, and it accounts for 10-20% of acute leukemias in adults. The prognosis for both adult and especially childhood ALL has improved substantially since the 1970s. The 5- year survival is approximately 95% in children. In adults, the 5-year survival varies between 25% and 75%, with more favorable results in younger than in older patients.
For more information about Acute Lymphoblastic Leukemia please visit our Leukemia Information Page
You need to be granted access to use the Acute Lymphoblastic Leukemia Image Database for Image Processing dataset. You can find the application form and information about getting access to the dataset on this page as well as information on how to contribute back to the project here. If you are not able to obtain a copy of the dataset please feel free to try this tutorial on your own dataset, we would be very happy to find additional AML & ALL datasets.
To get started follow the project documentation.
- Ubuntu Installation Guide
- Notebook Usage Guide
- NVIDIA Jetson Nano Installation Guide
- NVIDIA Jetson Nano Usage Guide
Asociación de Investigacion en Inteligencia Artificial Para la Leucemia Peter Moss encourages and welcomes code contributions, bug fixes and enhancements from the Github community.
Please read the CONTRIBUTING document for a full guide to forking our repositories and submitting your pull requests. You will also find our code of conduct in the Code of Conduct document.
- Adam Milton-Barker - Asociación de Investigacion en Inteligencia Artificial Para la Leucemia Peter Moss President/Founder & Lead Developer, Sabadell, Spain
We use SemVer for versioning.
This project is licensed under the MIT License - see the LICENSE file for details.
We use the repo issues to track bugs and general requests related to using this project. See CONTRIBUTING for more info on how to submit bugs, feature requests and proposals.