From 17580333b85a2d4ab6540883394751fee1a3273b Mon Sep 17 00:00:00 2001 From: kit <101046518@qq.com> Date: Fri, 26 Apr 2024 11:56:18 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/main' --- metagpt/environment/android/android_ext_env.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/metagpt/environment/android/android_ext_env.py b/metagpt/environment/android/android_ext_env.py index 8b9638fb7..bf64c0988 100644 --- a/metagpt/environment/android/android_ext_env.py +++ b/metagpt/environment/android/android_ext_env.py @@ -329,10 +329,10 @@ class AndroidExtEnv(ExtEnv): if iw > ih: x, y = y, x iw, ih = ih, iw - file_url = 'https://huggingface.co/ShilongLiu/GroundingDINO/blob/main/groundingdino_swint_ogc.pth' # 加载远程model - target_folder = Path(f'{DEFAULT_WORKSPACE_ROOT}/weights') - file_path = download_model(file_url, target_folder) if self.cv_model_status['clip_model_loaded'] == False: + file_url = 'https://huggingface.co/ShilongLiu/GroundingDINO/blob/main/groundingdino_swint_ogc.pth' # 加载远程model + target_folder = Path(f'{DEFAULT_WORKSPACE_ROOT}/weights') + file_path = download_model(file_url, target_folder) groundingdino_model = load_model(file_path, device=device).eval() self.cv_model_status['clip_model_loaded'] = True in_coordinate, out_coordinate = det(image, "icon", groundingdino_model) # 检测icon