This repository contains a Streamlit application for classifying OCT scan images using a deep learning model trained on a large dataset.
- Model Architecture: ResNet50
- Accuracy: 99.59%
- Training Data: 84,000 OCT scan images
Watch the demo video of the application on YouTube:
Make sure you have Python installed. You can download it from python.org.
-
Clone the repository:
git clone https://github.com/ramchandra06032004/OCT-images-classifier.git cd OCT-images-classifier
-
Create a virtual environment (optional but recommended):
python -m venv venv
-
Activate the virtual environment:
-
On Windows:
venv\Scripts\activate
-
On macOS and Linux:
source venv/bin/activate
-
-
Install the required packages individually:
-
Pillow:
pip install Pillow
-
NumPy:
pip install numpy
-
Pandas:
pip install pandas
-
TensorFlow:
pip install tensorflow
-
Streamlit:
pip install streamlit
-
Plotly:
pip install plotly
-
Scikit-learn:
pip install scikit-learn
-
Before running the application, download the trained model from Google Drive (link provided below) and place it in the ModelTraining
folder.
-
Navigate to the
streamlit_app
directory:cd streamlit_app
-
Run the Streamlit app:
streamlit run app.py
-
Open your web browser and go to
http://localhost:8501
to view the app.
- High Accuracy: Our model achieves an impressive 99.59% accuracy in classifying OCT scan images.
- Robust Architecture: Utilizes the powerful ResNet50 architecture, ensuring efficient and accurate image classification.
- Extensive Training Data: Trained on a comprehensive dataset of 84,000 OCT scan images, enhancing the model's reliability and performance.