Spaces:
Sleeping
Sleeping
slim-face-recognition/
βββ src/ # Source code for the project
β βββ slim_face/ # Main package for your project
β β βββ __init__.py # Marks directory as a Python package
β β βββ models/ # Model definitions and architectures
β β β βββ __init__.py
β β β βββ edgeface.py # Model definitions (e.g., edgeface backbones)
β β β βββ ... # Other model-related scripts
β β βββ data/ # Data loading and preprocessing
β β β βββ __init__.py
β β β βββ dataset.py # Custom Dataset classes for DataLoader
β β β βββ align.py # Face alignment utilities (e.g., from edgeface)
β β β βββ ... # Other data-related scripts
β β βββ training/ # Training-related scripts and logic
β β β βββ __init__.py
β β β βββ train.py # Main training script
β β β βββ accelerate_train.py # Accelerated training script
β β β βββ ... # Other training utilities
β β βββ inference/ # Inference-related scripts and logic
β β β βββ __init__.py # Marks directory as a Python package
β β β βββ inference.py # Face recognition inference logic
β β βββ utils/ # Utility functions (e.g., logging, metrics)
β β β βββ __init__.py
β β β βββ helpers.py # Miscellaneous helper functions
β β β βββ ... # Other utility scripts
β β βββ __main__.py # Entry point for running the package as a module
βββ tests/ # Unit and integration tests
β βββ __init__.py
β βββ test_data.py # Tests for data loading
β βββ test_models.py # Tests for model functionality
β βββ test_training.py # Tests for training pipeline
β βββ test_inference.py # Tests for inference pipeline
β βββ test_images/ # Sample images for testing (e.g., Elon_Musk.jpg)
βββ data/ # Datasets and data-related files
β βββ raw/ # Raw, unprocessed data
β βββ processed/ # Preprocessed data (e.g., aligned faces)
β βββ external/ # External datasets (e.g., from Kaggle)
βββ scripts/ # Standalone scripts for tasks like data download
β βββ download_dataset.py # Script to download datasets (e.g., Kaggle)
β βββ preprocess.py # Data preprocessing scripts
βββ notebooks/ # Jupyter notebooks for exploration and analysis
β βββ ztest.ipynb # Existing notebook for testing/exploration
β βββ ... # Other exploratory notebooks
βββ ckpts/ # Model checkpoints and weights
β βββ edgeface_xs_gamma_06.pt # Pretrained model weights
β βββ edgeface_s_gamma_05.pt # Pretrained model weights
β βββ ... # Other checkpoints
βββ configs/ # Configuration files (e.g., YAML, JSON)
β βββ training.yaml # Training hyperparameters
β βββ model.yaml # Model configurations
βββ docs/ # Documentation files
β βββ api.md # API documentation
β βββ usage.md # Usage instructions
βββ requirements.txt # Main dependencies
βββ requirements_compatible.txt # Development dependencies (e.g., testing, linting)
βββ README.md # Project overview and setup instructions
βββ LICENSE # License file (e.g., MIT, Apache)
βββ .gitignore # Git ignore file
βββ .python-version # Python version specification (e.g., for pyenv)
βββ setup.py # Setup script for packaging the project
βββ pyproject.toml # Modern Python project configuration (optional)