add model auto download

Author: zhiboniu

ppdiffusers/examples/controlnet/annotator/ppdet_hrnet/__init__.py

@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
-
 import cv2
 import numpy as np
 import paddle

ppdiffusers/examples/controlnet/annotator/ppdet_hrnet/det_keypoint_unite_infer.py

@@ -21,6 +21,8 @@
 import paddle
 import yaml
 
+from paddlenlp.utils.downloader import get_path_from_url_with_filelock
+
 from .det_keypoint_unite_utils import argsparser
 from .infer import (  # noqa F401
     Detector,
@@ -312,6 +314,15 @@ def exponential_smoothing(self, x_cur, x_pre, index=0):
 enable_mkldnn = False
 det_threshold = 0.4
 
+if not os.path.exists(det_model_dir):
+    detmodel_url = (
+        "https://bj.bcebos.com/v1/paddledet/models/keypoint/tinypose_enhance/picodet_s_320_lcnet_pedestrian.zip"
+    )
+    get_path_from_url_with_filelock(detmodel_url, root_dir="annotator/ppdet_hrnet/models/")
+if not os.path.exists(keypoint_model_dir):
+    kptmodel_url = "https://bj.bcebos.com/v1/paddledet/models/pipeline/dark_hrnet_w32_256x192.zip"
+    get_path_from_url_with_filelock(kptmodel_url, root_dir="annotator/ppdet_hrnet/models/")
+
 
 class PPDetPose(object):
     def __init__(self) -> None: