config.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
-------------------------------------------------
   @File Name:     config.py
   @Author:        Luyao.zhang
   @Date:          2023/5/16
   @Description: configuration file
-------------------------------------------------
"""
from pathlib import Path
import sys

# Get the absolute path of the current file
file_path = Path(__file__).resolve()

# Get the parent directory of the current file
root_path = file_path.parent

# Add the root path to the sys.path list if it is not already there
if root_path not in sys.path:
    sys.path.append(str(root_path))

# Get the relative path of the root directory with respect to the current working directory
ROOT = root_path.relative_to(Path.cwd())

print(ROOT)
# Source
SOURCES_LIST = ["Image", "Video", "Webcam"]


# DL model config
DETECTION_MODEL_DIR = ROOT / 'weights'
#print(DETECTION_MODEL_DIR)
YOLOv8n = DETECTION_MODEL_DIR / "yolov8n_char_new.pt"
YOLOv8s = DETECTION_MODEL_DIR / "best.pt"



DETECTION_MODEL_LIST = [
    "yolov8n.pt",
    "yolov8s.pt"]


OBJECT_COUNTER = None
OBJECT_COUNTER1 = None