HilmiZr commited on
Commit
53132c1
Β·
1 Parent(s): d9bd7b6

updated: models path on config

Browse files
Files changed (1) hide show
  1. config.py +3 -6
config.py CHANGED
@@ -11,12 +11,9 @@
11
  from pathlib import Path
12
 
13
  # DL model config
14
- DETECTION_MODEL_DIR = Path(__file__).resolve().parent / 'weights' / 'detection'
15
 
16
  DETECTION_MODEL_LIST = [
17
- "yolov8n.pt",
18
- "yolov8s.pt",
19
- "yolov8m.pt",
20
- "yolov8l.pt",
21
- "yolov8x.pt"
22
  ]
 
11
  from pathlib import Path
12
 
13
  # DL model config
14
+ DETECTION_MODEL_DIR = Path(__file__).resolve().parent / 'models'
15
 
16
  DETECTION_MODEL_LIST = [
17
+ "yolov8s-acnedetect-best.pt",
18
+ "yolov8m-acnedetect-best.pt"
 
 
 
19
  ]