diff --git a/install.py b/install.py index f46d9c9..5a55af7 100644 --- a/install.py +++ b/install.py @@ -19,8 +19,21 @@ if model_path is None: ultralytics_bbox_path = os.path.join(model_path, "ultralytics", "bbox") ultralytics_segm_path = os.path.join(model_path, "ultralytics", "segm") -if not os.path.exists(ultralytics_bbox_path): - os.makedirs(ultralytics_bbox_path) +if not os.path.exists(os.path.join(subpack_path, '..', '..', 'skip_download_model')): + if not os.path.exists(ultralytics_bbox_path): + os.makedirs(ultralytics_bbox_path) -if not os.path.exists(ultralytics_segm_path): - os.makedirs(ultralytics_segm_path) \ No newline at end of file + if not os.path.exists(ultralytics_segm_path): + os.makedirs(ultralytics_segm_path) + + if not os.path.exists(os.path.join(ultralytics_bbox_path, "face_yolov8m.pt")): + download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8m.pt", + ultralytics_bbox_path) + + if not os.path.exists(os.path.join(ultralytics_bbox_path, "hand_yolov8s.pt")): + download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/hand_yolov8s.pt", + ultralytics_bbox_path) + + if not os.path.exists(os.path.join(ultralytics_segm_path, "person_yolov8m-seg.pt")): + download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/person_yolov8m-seg.pt", + ultralytics_segm_path)