|
from pathlib import Path |
|
import sys |
|
|
|
|
|
file_path = Path(__file__).resolve() |
|
|
|
|
|
root_path = file_path.parent |
|
|
|
|
|
if root_path not in sys.path: |
|
sys.path.append(str(root_path)) |
|
|
|
|
|
ROOT = root_path.relative_to(Path.cwd()) |
|
|
|
|
|
IMAGE = 'Image' |
|
VIDEO = 'Video' |
|
WEBCAM = 'Webcam' |
|
RTSP = 'RTSP' |
|
YOUTUBE = 'YouTube' |
|
|
|
SOURCES_LIST = [IMAGE, VIDEO, WEBCAM, RTSP, YOUTUBE] |
|
|
|
|
|
IMAGES_DIR = ROOT / 'images' |
|
DEFAULT_IMAGE = IMAGES_DIR / 'office_4.jpg' |
|
DEFAULT_DETECT_IMAGE = IMAGES_DIR / 'office_4_detected.jpg' |
|
|
|
|
|
VIDEO_DIR = ROOT / 'videos' |
|
VIDEO_1_PATH = VIDEO_DIR / 'video_1.mp4' |
|
VIDEO_2_PATH = VIDEO_DIR / 'video_2.mp4' |
|
VIDEO_3_PATH = VIDEO_DIR / 'video_3.mp4' |
|
VIDEO_4_PATH = VIDEO_DIR / 'video_4.mp4' |
|
VIDEO_5_PATH = VIDEO_DIR / 'video_5.mp4' |
|
VIDEOS_DICT = { |
|
'video_1': VIDEO_1_PATH, |
|
'video_2': VIDEO_2_PATH, |
|
'video_3': VIDEO_3_PATH, |
|
'video_4': VIDEO_4_PATH, |
|
'video_5': VIDEO_5_PATH, |
|
} |
|
|
|
|
|
MODEL_DIR = ROOT / 'weights' |
|
DETECTION_MODEL = MODEL_DIR / 'yolov8n.pt' |
|
SEGMENTATION_MODEL = MODEL_DIR / 'yolov8n-seg.pt' |
|
|
|
|
|
WEBCAM_PATH = 0 |
|
|