DeepShield-Detectors / requirements.txt
AMontiB
Add utilities and CLI for PGC pipeline
c55595d
Raw History Blame Contribute Delete
645 Bytes
# ML / DL Frameworks
torch>=2.5.0
torchvision>=0.20.0
transformers>=4.57,<5 # PGC official weights need the transformers 4.x DINOv2 key names
# ML/DL Helpers
einops==0.8.0
# Data Science
# Using headless opencv to prevent libGL errors in Docker/Spaces
opencv-python-headless==4.10.0.84
pandas==2.2.3
scikit-image==0.22.0
scikit-learn==1.5.2
numpy<2
matplotlib
# MLOps & Infrastructure
wandb==0.16.6
# Utilities
PyYAML==6.0.2
tqdm==4.67.1
albumentations>=1.4.24,<2
kornia>=0.8,<0.9
scipy
huggingface_hub
ftfy
open-clip-torch
# 1. Use latest Gradio
gradio==5.49.1
# 2. Unpin FastAPI and Pydantic to allow latest versions
# fastapi
# pydantic