從零體檢一個魔塔社群模型(modelscope)最簡單demo

樱花落舞發表於2024-08-04

從社群拿一個模型,比如以下這個連結 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

從零體檢一個魔塔社群模型(modelscope)最簡單demo

其次配置Conda環境,選擇現有環境modelscope

從零體檢一個魔塔社群模型(modelscope)最簡單demo

如果以上步驟都順利,此時在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': ['正常', '綠屏', '破圖']}

相關文章