update

preprocess/openpose/annotator/openpose/__init__.py

@@ -17,6 +17,7 @@ from .hand import Hand
 from .face import Face
 from annotator.util import annotator_ckpts_path
 import requests
+import shutil
 
 body_model_path = "https://huggingface.co/lllyasviel/Annotators/resolve/main/body_pose_model.pth"
 hand_model_path = "https://huggingface.co/lllyasviel/Annotators/resolve/main/hand_pose_model.pth"
@@ -45,18 +46,34 @@ def draw_pose(pose, H, W, draw_body=True, draw_hand=True, draw_face=True):
 
 class OpenposeDetector:
     def __init__(self):
-        body_modelpath = "/home/user/app/checkpoints/openpose/ckpts/body_pose_model.pth"
+        body_modelpath = os.path.join(annotator_ckpts_path, "body_pose_model.pth")
         # hand_modelpath = os.path.join(annotator_ckpts_path, "hand_pose_model.pth")
         # face_modelpath = os.path.join(annotator_ckpts_path, "facenet.pth")
-
+        #判断body_modelpath是否是一个目录
+        if os.path.isdir(body_modelpath):
+            print("body_modelpath是一个目录")
+            #删除目录
+            shutil.rmtree(body_modelpath)
+            print("body_modelpath目录删除完成")
+            
+
+        if not os.path.exists(annotator_ckpts_path):
+            #创建目录
+            os.makedirs(annotator_ckpts_path)
+            #从url下载文件
+        response = requests.get(body_model_path)
+        with open(body_modelpath, "wb") as f:
+            f.write(response.content)
+        print("body_pose_model.pth下载完成")
+
+        #判断文件是否存在
         if not os.path.exists(body_modelpath):
+            print("body_pose_model.pth下载失败")
+        
+            
+
             # from basicsr.utils.download_util import load_file_from_url
             # load_file_from_url(body_model_path, model_dir=annotator_ckpts_path)
-            response = requests.get(body_model_path)
-            #下载该文件
-            with open("body_pose_model.pth", "wb") as f:
-                f.write(response.content)
-            print("body_pose_model.pth下载完成")
 
         # if not os.path.exists(hand_modelpath):
         #     from basicsr.utils.download_util import load_file_from_url