shape-classifier / config.py
immortalcunt's picture
Upload 5 files
d6e806a verified
raw
history blame contribute delete
774 Bytes
import os
from pathlib import Path
class Config:
# Base directory
BASE_DIR = Path(__file__).parent
# Flask configuration
SECRET_KEY = os.environ.get('SECRET_KEY') or 'your-secret-key-here'
MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
# Directory configuration
UPLOAD_FOLDER = BASE_DIR / 'static' / 'uploads'
CLASSIFIED_FOLDER = BASE_DIR / 'static' / 'classified_results'
ANNOTATIONS_FOLDER = BASE_DIR / 'static' / 'annotations'
# Image processing configuration
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
MIN_CONTOUR_AREA = 100
DEFAULT_SIMILARITY_THRESHOLD = 0.85
# Feature extraction configuration
USE_ADVANCED_FEATURES = True
TEXTURE_FEATURES_ENABLED = True
HU_MOMENTS_ENABLED = True