This project implements a deep learning Convolutional Neural Network (CNN) to classify medical organ images into four categories: Heart, Brain, Liver, Limbs.
- π½οΈ Video
- π Project Structure
- π οΈ Prerequisites
- β¨ Key Features
- π οΈ Installation
- π Dataset Preparation
- π§ Model Architecture
- π Training
- πͺ Usage
- π Performance Metrics
- πΎ Model Saving
- π οΈ Customization
β οΈ Limitations- π Future Improvements
- π€ Contributions
- π License
- π§ Contact
Video1.mp4
project_root/
β
βββ data_sets/
β βββ heart/
β βββ brain/
β βββ liver/
β βββ limbs/
β
βββ test_data/
β βββ NONE/
β
βββ save_model/
β βββ organsClssify.h5
β
βββ logs/
- Python 3.8+
- TensorFlow
- Numpy
- Matplotlib
- Image classification using CNN
- Data augmentation
- TensorBoard logging
- Model saving and loading functionality
- Visulization of training metrics
- Clone the repository
- Install required dependencies:
pip install tensorflow numpy matplotlib- Organize images in
data_sets/directory - Each subdirectory represents a different organ class
- Recommended image size: 256x256 pixels
- Convolutional layers for feature extraction
- MaxPooling for dimensionality reduction
- Dropout for preventing overfitting
- Dense layers for classification
- Uses Adam optimizer
- Categorical cross-entropy loss
- 5 training epochs
- Automatic train/validation/test split
# Run the main script to train the model
python organ_classification.pytest_model('path/to/test_image.jpg')- Training and validation loss tracked
- Training and validation accuracy visualized
- Final test set evaluation provided
- Trained model saved as
organsClssify.h5 - Can be loaded for future predictions
- Adjust hyperparameters in the script
- Modify model architecture as needed
- Requires diverse and well-labeled training data
- Performance depends on image quality and dataset composition
- Increase training epochs
- Implement more advanced data augmentation
- Experiment with deeper network architectures
Contributions are welcome! Feel free to fork this repository, create a branch, and submit a pull request with your enhancements.
This project is licensed under the MIT License - see the LICENSE file for details.
If you have any questions or suggestions, feel free to reach out via GitHub or email at [Ibrahim.Ibrahim051@eng-st.cu.edu.eg]