This project is a deep learning-based gender detection system using PyTorch. It includes model training, inference pipeline, and testing on images. The project utilizes facial features to predict the gender of individuals in images.
Gender_Detection_System/
├── main.ipynb # Main logic and pipeline for gender detection
├── Pipeline.ipynb # Additional pipeline steps
├── best_Gender.pt # Trained gender classification model
├── best_Classifier.pt # Another model (e.g., classifier for feature extraction)
├── Testing_Images/ # Sample images for testing the model
git clone https://github.com/Muhamad-Usman55/Gender_Detection_System.git
cd Gender_Detection_Systempip install -r requirements.txtOpen main.ipynb in Jupyter Notebook and follow the steps to test gender prediction.
- torch
- torchvision
- matplotlib
- numpy
- opencv-python
- PIL
best_Gender.pt: Pre-trained gender detection model.best_Classifier.pt: Auxiliary model (could be feature extractor or classifier).
Located in Testing_Images/ folder.
This project is licensed under the MIT License - see the LICENSE file for details.