config.py

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

# DL model config
DETECTION_MODEL_DIR = Path(__file__).resolve().parent / 'models'

DETECTION_MODEL_LIST = [
    "yolov8s-acnedetect-best.pt",
    "yolov8m-acnedetect-best.pt"
]