This guide will take you through using the ALL Jetson Nano Classifier Jupyter Notebook to train and test your classifier.
First you need to install the required software for training the model. Below are the available installation guides:
We will build a Convolutional Neural Network with the following architecture:
- Average pooling layer
- Conv layer
- Depthwise conv layer
- Flatten layer
- Fully connected layer
- Softmax layer
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.
Once you have your data you need to add it to the project filesystem. You will notice the data folder in the Model directory, model/data, inside you have train & test. Add all of the images from the ALL_IDB1 dataset to the model/data/train folder.
We will create an augmented dataset based on the Leukemia Blood Cell Image Classification Using Convolutional Neural Network by T. T. P. Thanh, Caleb Vununu, Sukhrob Atoev, Suk-Hwan Lee, and Ki-Ryong Kwon.
In the data processing stage, ten negative images and ten positive images are removed from the dataset and moved to the model/data/test/ directory. This data is not seen by the network during the training process, and is used to test the performance of the model.
To ensure your model gets the same results, you should use the same test images. You can also try with your own image selection, however results may vary.
Now you need to start Jupyter Notebooks. In your project root execute the following command, replacing the IP and port as desired.
jupyter notebook --ip YourIP --port 8888
Navigate to the URL provided when starting Jupyter Notebooks and you should be in the project root. Now navigate to notebooks/classifier.ipynb. With everything set up you can now begin training. Run the Jupyter Notebook and wait for it to finish.
After training the model was converted to ONNX format ready to be converted to TensorRT format for usage on the NVIDIA Jetson Nano.
Here we trained a deep learning model for Acute Lymphoblastic Leukemia detection utilizing Intel® Optimization for Tensorflow* from the Intel® oneAPI AI Analytics Toolkit to optimize and accelarate training. We introduced a 6 layer deep learning model and applied data augmentation to increase the training data.
We trained our model with a target of 150 epochs and used early stopping to avoid overfitting. The model trained for 37 epochs resulting in a fairly good fit, accuracy/precision/recall and AUC are satisfying. In addition the model reacts well during testing classifying each of the twenty unseen test images correctly.
We see that the TFRT model performs considerably better than the Tensorflow model, improving classification time by 0.35 seconds.
- To test with the Python scripts continue to the Python usage documentation
- To begin installation on your Jetson Nano continue to the Jetson Nano installation documentation
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.