修改工具类查找相对路径函数名称

watermark_generate/deals/yolox_pytorch_black_embed.py

@@ -14,7 +14,7 @@ def modify_model_project(secret_label: str, project_dir: str, public_key: str):
     # 对密码标签进行切分，根据密码标签长度，目前进行三等分
     secret_parts = general_tool.divide_string(secret_label, 3)
 
-    rela_project_path = general_tool.find_yolox_directories(project_dir, 'YOLOX')
+    rela_project_path = general_tool.find_relative_directories(project_dir, 'YOLOX')
     if not rela_project_path:
         raise BusinessException(message="未找到指定模型的工程目录", code=-1)
 

watermark_generate/tools/general_tool.py

@@ -22,7 +22,7 @@ def divide_string(s, num_parts):
     return parts
 
 
-def find_yolox_directories(root_dir, target_dir):
+def find_relative_directories(root_dir, target_dir):
     """
     查找指定目录下的目标目录相对路径
     :param root_dir: 根目录
@@ -32,7 +32,7 @@ def find_yolox_directories(root_dir, target_dir):
     root_path = Path(root_dir)
     yolox_paths = []
 
-    # 递归查找名为 'yolox' 的目录
+    # 递归查找指定目录
     for path in root_path.rglob(target_dir):
         if path.is_dir():
             # 计算相对路径