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}")