config.py

"""
全局配置 —— 集中管理所有路径常量和算法参数
"""
import os

ROOT_DIR = os.path.dirname(os.path.abspath(__file__))

# ==================== 目录 ====================
RESOURCES_DIR = os.path.join(ROOT_DIR, 'resources')
ICONS_DIR = os.path.join(RESOURCES_DIR, 'icons')
STYLES_DIR = os.path.join(RESOURCES_DIR, 'styles')
REFERENCE_DIR = os.path.join(RESOURCES_DIR, 'reference')
MODELS_DIR = os.path.join(ROOT_DIR, 'models')
DATA_DIR = os.path.join(ROOT_DIR, 'data')
OUTPUT_DIR = os.path.join(DATA_DIR, 'output')

# ==================== 关键文件路径 ====================
DATABASE_PATH = os.path.join(DATA_DIR, 'database.db')
CAPTURE_PATH = os.path.join(DATA_DIR, 'capture.png')
QSS_PATH = os.path.join(STYLES_DIR, 'style.qss')
APP_ICON_PATH = os.path.join(ICONS_DIR, 'app.ico')

# ==================== 图标路径 ====================
ICON_LEFT_ROTATE = os.path.join(ICONS_DIR, 'left_rotate.png')
ICON_RIGHT_ROTATE = os.path.join(ICONS_DIR, 'right_rotate.png')
ICON_HISTOGRAM = os.path.join(ICONS_DIR, 'histogram.png')
ICON_COLOR = os.path.join(ICONS_DIR, 'color.png')

# ==================== YOLOv5 检测参数 ====================
DETECTION_AREA_THRESHOLD = 0.755

# ==================== Retinex 增强参数 ====================
RETINEX_SIGMA_LIST = [15, 80, 200]
RETINEX_LOW_CLIP = 0.01
RETINEX_HIGH_CLIP = 0.99

# ==================== 像素换算比例 ====================
PIXEL_TO_CM = 1.0 / 50
PIXEL_AREA_TO_CM2 = 1.0 / (225 * 225)