weapon-detection-app / .env.example
Marcus Vinicius Zerbini Canhaço
feat: atualização do detector com otimizações para GPU T4
346e896
# Configurações da Aplicação
DEBUG=False
PORT=7860
HOST=0.0.0.0
# Configurações do Modelo
MODEL_THRESHOLD=0.4
TARGET_FPS=2
# Configurações de Cache
CACHE_DIR=/tmp/weapon_detection_cache
MAX_CACHE_SIZE=1000000000 # 1GB em bytes
RESULT_CACHE_SIZE=1000
# Configurações de Upload
MAX_CONTENT_LENGTH=100000000 # 100MB em bytes
ALLOWED_EXTENSIONS=mp4,avi,mov,webm
# Configurações de Processamento
MAX_VIDEO_DURATION=300 # 5 minutos em segundos
MIN_CONFIDENCE=0.3
MAX_DETECTIONS_PER_FRAME=10
# Configurações de E-mail
EMAIL_APP_PASSWORD=sua_senha_de_app_aqui
NOTIFICATION_EMAIL="" # E-mail para envio de notificações
# Configurações do Modelo
HUGGING_FACE_TOKEN="" # Token do Hugging Face para acesso aos modelos
TOKENIZERS_PARALLELISM=false
MODEL_CACHE_DIR=./.model_cache
MAX_WORKERS=2
USE_HALF_PRECISION=true
DETECTION_CONFIDENCE_THRESHOLD=0.5
MODEL_CONFIDENCE_THRESHOLD=0.5
MODEL_IOU_THRESHOLD=0.45
# Configurações do Servidor
SERVER_HOST=0.0.0.0
SERVER_PORT=7860
ENABLE_SHARING=true # true para ambiente local, false para Hugging Face
# Configurações de Vídeo
DEFAULT_FPS=2
DEFAULT_RESOLUTION=640
# Configurações de GPU (opcional - apenas se tiver GPU)
CUDA_VISIBLE_DEVICES=0
TORCH_CUDA_ARCH_LIST="7.5"
NVIDIA_VISIBLE_DEVICES=all
NVIDIA_DRIVER_CAPABILITIES=compute,utility
# Configurações do modelo
MODEL_NAME=google/owlv2-base-patch16
DEVICE=cuda:0
THRESHOLD=0.3
FPS=2
RESOLUTION=640
# Configurações de logging
LOG_LEVEL=INFO
LOG_FORMAT=%(asctime)s - %(name)s - %(levelname)s - %(message)s
# Configurações de memória
MAX_MEMORY=8GB
OFFLOAD_FOLDER=offload
# Configurações de otimização
TORCH_COMPILE=false
HALF_PRECISION=true
# Configurações de interface
THEME=default
ALLOW_FLAGGING=false
ALLOW_SCREENSHOT=true
SHOW_API=true
# Configurações de segurança
AUTHENTICATION=false
USERNAME=admin
PASSWORD=admin
# Configurações de monitoramento
ENABLE_QUEUE=true
MAX_THREADS=4
CONCURRENCY_COUNT=1
# Configurações de armazenamento
SAVE_OUTPUTS=false
OUTPUT_DIR=outputs