File size: 1,220 Bytes
6a3a328 d03afc0 6a3a328 96999c2 5da8fea 96999c2 5da8fea 6a3a328 5da8fea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
#!/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())
# Source
SOURCES_LIST = ["Image"]
# DL model config
#DETECTION_MODEL_DIR = ROOT / 'models' / 'detection'
DETECTION_MODEL_DIR = ROOT
#YOLOv8n = DETECTION_MODEL_DIR / "yolov8n.pt"
YOLOv8s = 'crowded_human_yolov8s.pt'
#YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
#YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
#YOLOv8x = DETECTION_MODEL_DIR / "yolov8x.pt"
DETECTION_MODEL_LIST = [
# "yolov8n.pt",
"yolov8s.pt"]
# "yolov8m.pt",
# "yolov8l.pt",
# "yolov8x.pt"]
|