bmy/python-ai-video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改一些文件

Browse Source
This commit is contained in:
编码猿
2025-01-21 19:00:59 +08:00
parent 275c6c4cec
commit b083368f67
7 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@@ -2,6 +2,9 @@
接口文档：http://127.0.0.1:8000/docs。 接口文档：http://127.0.0.1:8000/docs。
AI模型的本地地址 存储路径：/Users/bmy/.cache/modelscope/hub/damo
fairseq 依赖 需要 windows 先安装 vs_BuildTools.exe fairseq 依赖 需要 windows 先安装 vs_BuildTools.exe
decord 用 eva-decord 替代 decord 用 eva-decord 替代

2
requirements.txt
View File

@@ -3,7 +3,7 @@ uvicorn==0.27.1
pydantic==2.6.1 pydantic==2.6.1
pydantic-settings==2.2.0 pydantic-settings==2.2.0
requests==2.31.0 requests==2.31.0
numpy==1.26.4
torch==2.2.0 torch==2.2.0
opencv-python==4.9.0.80 opencv-python==4.9.0.80
open-clip-torch==2.24.0 open-clip-torch==2.24.0

2
src/ai/ai_test.py
View File

@@ -4,7 +4,7 @@ from modelscope.pipelines import pipeline
from modelscope.preprocessors.image import load_image 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') 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 = ["小火龙", "白色衬衫"] input_texts = ["小火龙", "白色衬衫"]
# 支持一张图片(PIL.Image)或多张图片(List[PIL.Image])输入，输出归一化特征向量 # 支持一张图片(PIL.Image)或多张图片(List[PIL.Image])输入，输出归一化特征向量

2
src/ai/multimModaImageTextSearch.py
View File

@@ -5,7 +5,7 @@ from modelscope.preprocessors.image import load_image
from fastapi import WebSocket from fastapi import WebSocket
class multimModaImageTextSearch: class multimModaImageTextSearch:
# AI模型的本地地址 # AI模型的本地地址 存储路径：/Users/bmy/.cache/modelscope/hub/damo
modelName: str = 'damo/multi-modal_clip-vit-large-patch14_336_zh' modelName: str = 'damo/multi-modal_clip-vit-large-patch14_336_zh'
# 使用 cpu 还是 GPU 计算，GPU 比 CPU快，但是 GPU 只支持 英伟达 显卡，至少4G显存以上 # 使用 cpu 还是 GPU 计算，GPU 比 CPU快，但是 GPU 只支持 英伟达 显卡，至少4G显存以上
deviceType: str = 'cpu' # cpu, gpu deviceType: str = 'cpu' # cpu, gpu

1
src/celery/ai_image/task.py
View File

@@ -27,6 +27,7 @@ def start_ai_img(self, keyWord: str, imgList: list[str]):
key=keyWord, key=keyWord,
imgList=imgList imgList=imgList
) )
print("imgList: ", imgList)
res = image.start() res = image.start()
return res return res

2
src/http/_config.py
View File

@@ -6,7 +6,7 @@ class HttpConfig:
'Host': 'www.kuaishou.com', 'Host': 'www.kuaishou.com',
'Referer': 'https://www.kuaishou.com/profile/3x2347wkfukzx4g', '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', '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 { visionProfileLikePhotoList: str = ''' fragment photoContent on PhotoEntity {
__typename __typename

2
test/index.html
View File

@@ -41,7 +41,7 @@
<script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.7.1/jquery.min.js"></script> <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.7.1/jquery.min.js"></script>
<script src="https://cdn.bootcdn.net/ajax/libs/axios/1.5.0/axios.js"></script> <script src="https://cdn.bootcdn.net/ajax/libs/axios/1.5.0/axios.js"></script>
<script> <script>
let pcursorNum = '1703970033000', feedsList = [], filterList = [], intervalId = 0; let pcursorNum = '', feedsList = [], filterList = [], intervalId = 0;
axios.defaults.baseURL = 'http://127.0.0.1:8000/api/v1'; axios.defaults.baseURL = 'http://127.0.0.1:8000/api/v1';
axios.interceptors.response.use(response => response.data, error => Promise.reject(error)); axios.interceptors.response.use(response => response.data, error => Promise.reject(error));