| # This file lists the required packages for the clip-face-attribute-classifier project. | |
| # Install them using: pip install -r requirements.txt | |
| # --- Hugging Face Libraries --- | |
| # Core library for models, Trainer, TrainingArguments, and processors | |
| transformers==4.38.2 | |
| # Used for data handling and creating Dataset objects | |
| datasets==2.18.0 | |
| # For efficient training and hardware acceleration with the Trainer | |
| accelerate==0.27.2 | |
| # For interacting with the Hugging Face Hub (login, upload, etc.) | |
| huggingface_hub==0.21.4 | |
| # --- Core Deep Learning Framework --- | |
| # The fundamental deep learning library | |
| torch==2.2.1 | |
| # Companion library for computer vision tasks in PyTorch | |
| torchvision==0.17.1 | |
| # --- Data Handling and Metrics --- | |
| # For reading and manipulating the .csv label files | |
| pandas==2.2.1 | |
| # For calculating evaluation metrics like accuracy, precision, recall, and F1-score | |
| scikit-learn==1.4.1.post1 | |
| # --- Utilities --- | |
| # For opening and handling image files | |
| Pillow==10.2.0 | |
| # For creating progress bars during evaluation | |
| tqdm==4.66.2 | |
| # For loading the safer .safetensors model format | |
| safetensors==0.4.2 | |