Full Project Demo: https://www.loom.com/share/434f009b5d61432da6ca36f963c73f98
This Jupyter Notebook provides a deep learning model for classifying pathology images using TensorFlow. The model is designed to work with three different datasets, which need to be organized into training, testing, and validation sets before running the model.
There are three separate components for this project:
- ML Model: https://github.com/VSTARprojects/pc_ml
- FrontEnd: https://github.com/VSTARprojects/pc_fe
- BackEnd: https://github.com/VSTARprojects/pc_dj
Before running the notebook, you will need the following:
- A working installation of TensorFlow https://www.tensorflow.org/install
- Download the required datasets. We have used the following:
- Install Jupyter Notebook https://jupyter.org/install
- Download the Jupyter Notebook and open it in your preferred notebook environment.
- Organize your datasets into separate directories for training, testing, and validation. The directories should be named train, test, and valid, respectively.
- Within each of these directories, create subdirectories for each class of pathology that you want to classify. For example, "colon_adenocarinoma", "colon_normal", "renal_clearcell" and so on...
- Place the appropriate images in each subdirectory.
- Modify the code in the notebook to point to the correct directories and set any necessary hyperparameters.
- Run
pip install -r requirements.txt - Run the notebook and wait for the model to finish training.
It is important that the datasets are organized correctly in order for the model to work properly. Each image should be in its own file and each file should be named uniquely, preferably with a sequential number.