File size: 1,216 Bytes
08d6e3c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from pathlib import Path
import sys


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", "Video", "Webcam"]


# DL model config
DETECTION_MODEL_DIR = ROOT / 'weights'/'Detection'
#YOLOv8n = DETECTION_MODEL_DIR / "yolov8n.pt"
#YOLOv8s = DETECTION_MODEL_DIR / "yolov8s.pt"
#YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
#YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
#YOLOv8x = DETECTION_MODEL_DIR / "yolov8x.pt"
YOLOV8custom1 = DETECTION_MODEL_DIR /"Rust.pt"
YOLOV8custom2 = DETECTION_MODEL_DIR / "Scratch.pt"
YOLOV8custom3 = DETECTION_MODEL_DIR / "Fabric.pt"

# If You Want To Use This Pre Train Model Then UnComment It and  uplode in your weights/detection dir and Use Them All

DETECTION_MODEL_LIST = [
    #"yolov8n.pt",
    #"yolov8s.pt",
    #"yolov8m.pt",
    #"yolov8l.pt",
    #"yolov8x.pt",
    "Rust.pt",
    "Scratch.pt",
    "Fabric.pt"]