def get_all_weights(weights_path="./weights/detect"):
    from pathlib import Path

    # 获取文件夹下所有文件（不包括文件夹）
    all_weights = [
        str(f.stem)
        for f in Path(weights_path).iterdir()
        if Path(f).is_file() and Path(f).suffix == ".pth"
    ]
    # print_info(all_weights)
    return all_weights


def now_time():
    from datetime import datetime

    return datetime.now().strftime(r"%Y-%m-%dT%H%M%S")


def print_info(text: str):
    BEGIN_COLOR = "\033[32m"
    END_COLOR = "\033[0m"
    print(f"{BEGIN_COLOR}{text}{END_COLOR}")