diff --git a/README.md b/README.md index 46d2a20..b9fe78a 100644 --- a/README.md +++ b/README.md @@ -2,6 +2,9 @@ 接口文档:http://127.0.0.1:8000/docs。 + +AI模型的本地地址 存储路径:/Users/bmy/.cache/modelscope/hub/damo + fairseq 依赖 需要 windows 先安装 vs_BuildTools.exe decord 用 eva-decord 替代 diff --git a/requirements.txt b/requirements.txt index 7100cf0..5995093 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ uvicorn==0.27.1 pydantic==2.6.1 pydantic-settings==2.2.0 requests==2.31.0 - +numpy==1.26.4 torch==2.2.0 opencv-python==4.9.0.80 open-clip-torch==2.24.0 diff --git a/src/ai/ai_test.py b/src/ai/ai_test.py index ecac21e..6efe8b4 100644 --- a/src/ai/ai_test.py +++ b/src/ai/ai_test.py @@ -4,7 +4,7 @@ from modelscope.pipelines import pipeline from modelscope.preprocessors.image import load_image pipeline = pipeline(task=Tasks.multi_modal_embedding, model='damo/multi-modal_clip-vit-large-patch14_336_zh', model_revision='v1.0.1', device='cpu') -input_img = load_image('https://p1.a.yximgs.com/upic/2023/12/27/13/BMjAyMzEyMjcxMzAwMzdfMjkwOTEwNDc5N18xMjA2NTYyODIyNDRfMl8z_Be33b730fcb8aaf000ba5e58bf8c3681e.jpg?tag=1-1708460644-unknown-0-huswbkiuiq-0ac88680ace16101&clientCacheKey=3x9ghy3dj8y2z5a.jpg&di=70208a94&bp=14764') # 支持皮卡丘示例图片路径/本地图片 返回PIL.Image +input_img = load_image('https://p2.a.yximgs.com/upic/2023/10/29/18/BMjAyMzEwMjkxODA4NTRfMTUxNDU3NTMwNV8xMTYwNDQ3MzU3OThfMV8z_B903dcf4acbd7359157de400894a486b2.jpg?tag=1-1737455669-unknown-0-eevfhc2ows-f3b84d4c8f01cc99&clientCacheKey=3x49fsxn85ijp6q.jpg&di=JA4DYiIS6gDJR6IE5Fh9Mg==&bp=14764') # 支持皮卡丘示例图片路径/本地图片 返回PIL.Image input_texts = ["小火龙", "白色衬衫"] # 支持一张图片(PIL.Image)或多张图片(List[PIL.Image])输入,输出归一化特征向量 diff --git a/src/ai/multimModaImageTextSearch.py b/src/ai/multimModaImageTextSearch.py index 8a7ede6..90d978b 100644 --- a/src/ai/multimModaImageTextSearch.py +++ b/src/ai/multimModaImageTextSearch.py @@ -5,7 +5,7 @@ from modelscope.preprocessors.image import load_image from fastapi import WebSocket class multimModaImageTextSearch: - # AI模型的本地地址 + # AI模型的本地地址 存储路径:/Users/bmy/.cache/modelscope/hub/damo modelName: str = 'damo/multi-modal_clip-vit-large-patch14_336_zh' # 使用 cpu 还是 GPU 计算,GPU 比 CPU快,但是 GPU 只支持 英伟达 显卡,至少4G显存以上 deviceType: str = 'cpu' # cpu, gpu diff --git a/src/celery/ai_image/task.py b/src/celery/ai_image/task.py index 014afaf..c1da74e 100644 --- a/src/celery/ai_image/task.py +++ b/src/celery/ai_image/task.py @@ -27,6 +27,7 @@ def start_ai_img(self, keyWord: str, imgList: list[str]): key=keyWord, imgList=imgList ) + print("imgList: ", imgList) res = image.start() return res diff --git a/src/http/_config.py b/src/http/_config.py index e897b59..78d38cf 100644 --- a/src/http/_config.py +++ b/src/http/_config.py @@ -6,7 +6,7 @@ class HttpConfig: 'Host': 'www.kuaishou.com', 'Referer': 'https://www.kuaishou.com/profile/3x2347wkfukzx4g', 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36', - 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_fca770657d0a24abb53ddc92bb42e612; didv=1700407978000; userId=58344290; kuaishou.server.web_st=ChZrdWFpc2hvdS5zZXJ2ZXIud2ViLnN0EqABFs7urbq3vvcrvB-IRBSKHXfdVRzblSFx7udepTzeAtgCompa6-aZiv1f_Bb66SubMkCbLODJQ3cn_GxlWWWbPgbQEyul9f7SDKsGphyvK1Pk3aj-yn899D9gR-xj1wiqT_0nA9x3XcPmvuJ8qQzIeLl3236npeMV0i9uYEAeIHDp9RtvPlQ4o0piktgGjMZH13Uhk7-ZfqDuUPQ-mEH7axoSS2UWrTnDahVDKzRYjzjLpJM-IiDXKnEX31rxM9oVovhBLMPSocxSvtN7wyo1oS1nNdvjSCgFMAE; kuaishou.server.web_ph=5b6a8c20ee92347d220cbc98f705f92042fb; kpn=KUAISHOU_VISION' + 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_1debac63b767ac564381b3e54dc1de14; userId=58344290; kuaishou.server.webday7_st=ChprdWFpc2hvdS5zZXJ2ZXIud2ViZGF5Ny5zdBKwAcl8si24RTFngM_Km0UdhjDwUfscFSn1oUV5P7SNRrD_XcCr2kca2dnV8IhO9Pn_AJ5hCLjDzEnK-2c4NKzFPfTghZop4VhZh_Xads52aoOZKXvdPJ5UIdOWkpI3IUpKKEj_skdwHAYczNqCgJA40-KFLjmF6PGUcH9F924l68rOy_rVDu912xf5zrm3V_2vXzv61lg9b2Y55uAuVm_GAJr8wO91scf8FjB3FWiz2zU-GhLcWLZU0wMn7F5ONAeuFgyQkCwiIF1D6aMj4ItLQrCzkiRZXt3U4o0QfkxNv8s9PzXJ3YGtKAUwAQ; kuaishou.server.webday7_ph=2ec00e134011cd2f1ecc4904954410a101be; kpn=KUAISHOU_VISION' } visionProfileLikePhotoList: str = ''' fragment photoContent on PhotoEntity { __typename diff --git a/test/index.html b/test/index.html index 5a8d985..2b9ab56 100644 --- a/test/index.html +++ b/test/index.html @@ -41,7 +41,7 @@