This project implements a body part recognition system using machine learning techniques. It processes DICOM images, converts them to PNG format, and utilizes SVM (Support Vector Machine) and KNN (K-Nearest Neighbors) classifiers to identify various body parts from medical images.
This project is based on my other project TrafficSignRecognition.
- Model Training: Train classification models using SVM and KNN algorithms.
- Prediction: Make predictions on test datasets and evaluate model performance.
- Data Handling: Load training and testing data from specified directories.
- DICOM Image Processing: Load and preprocess DICOM images for analysis.
- C++ Compiler (e.g., g++)
- CMake
- OpenCV
- DCMTK (for DICOM file handling)
-
Clone the repository:
git clone https://github.com/amWRit/BodyPartRecognition.gitcd BodyPartRecognition
-
Install the required dependencies (OpenCV, DCMTK).
- Prepare your dataset:
- Place your training data in the
bodyparts-train-datadirectory. - Place your test png images in the
imagesdirectory. - Place your test dicom images in the
dicom-imagesdirectory.
-
Run the application:
./BodyPartRecognition -
The application will load the data, train the models, and output predictions along with accuracy metrics.
- SVM (Support Vector Machine): A supervised learning model used for classification tasks.
- KNN (K-Nearest Neighbors): A simple, instance-based learning algorithm used for classification.
This project is licensed under the MIT License - see the LICENSE file for details.