从零体检一个魔塔社区模型(modelscope)最简单demo
从社区拿一个模型,比如以下这个链接 https://www.modelscope.cn/models/iic/cv_mobilenet-v2_bad-image-detecting
它的代码样例如下
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.outputs import OutputKeys
img = 'https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/dogs.jpg'
test_pipeline = pipeline(Tasks.bad_image_detecting, 'damo/cv_mobilenet-v2_bad-image-detecting')
result = test_pipeline(img)
print(result)
怎么在自己电脑环境中运行起来?
前期准备
首先安装一个Python环境,选最新版本安装就好了,然后把Python配置到环境变量。
按照官方文档的说法,需要安装一个conda环境,这里建议是安装一个miniconda版本,链接 Miniconda — Anaconda documentation
如果想要在cmd命令行使用,还可以将安装路径miniconda3\condabin
配置到环境变量
最后在控制台输入 conda -V
,能顺利显示版本号说明成功了。
C:\Users\Admin>conda -V
conda 24.5.0
创建一个conda环境
注意官方文档推荐使用python版本是3.8(如果提示没有权限Permission denied,那么使用管理员身份运行)
conda create --name modelscope python==3.8
通过 conda info -e
检查是否存在modelscope环境
C:\Windows\System32>conda info -e
# conda environments:
#
base C:\ProgramData\miniconda3
modelscope C:\ProgramData\miniconda3\envs\modelscope
在PyCharm使用conda环境
添加Python解释器 - Virtualenv环境,选择miniconda3\envs\modelscope
下的Python.exe
其次配置Conda环境,选择现有环境modelscope
如果以上步骤都顺利,此时在PyCharm终端看到的Python环境应该是
(modelscope) PS E:\GitHub\blogsCodeSimple\新建文件夹\modelscopeDemo> python
Python 3.8.0 (default, Nov 6 2019, 16:00:02) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
然后拉取 modelscope[framework]
和 opencv-python
点击运行即可(如果提示没有权限Permission denied,那么使用管理员身份运行PyCharm)
最后输出如下
{'scores': [1.0, 2.5564614e-10, 3.6810602e-13], 'labels': ['正常', '绿屏', '花屏']}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· .NET10 - 预览版1新功能体验(一)