| 123456789101112131415161718192021222324252627 |
- from pathlib import Path
- ROOT = Path(__file__).parent.parent
- RESOURCES_DIR = ROOT / "resources"
- WATER_MARK_TEMPLATE_IMAGE_PATH = RESOURCES_DIR / "watermark_template.png"
- WATER_MARK_DETECT_YOLO_WEIGHTS = RESOURCES_DIR / "best.pt"
- OUTPUT_DIR = ROOT / "output"
- OUTPUT_DIR.mkdir(exist_ok=True, parents=True)
- DEFAULT_WATERMARK_REMOVE_MODEL = "cv2"
- WORKING_DIR = ROOT / "working_dir"
- WORKING_DIR.mkdir(exist_ok=True, parents=True)
- LOGS_PATH = ROOT / "logs"
- LOGS_PATH.mkdir(exist_ok=True, parents=True)
- DATA_PATH = ROOT / "data"
- DATA_PATH.mkdir(exist_ok=True, parents=True)
- SQLITE_PATH = DATA_PATH / "db.sqlite3"
|
