基于modelscope搭建OCR大模型,在线部署及离线部署讲解

1、首先安装环境

安装环境教程:https://modelscope.cn/docs/%E7%8E%AF%E5%A2%83%E5%AE%89%E8%A3%85
我是直接拉取的GPU镜像,嗷嗷一顿干

docker pull registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.13.1

2、环境安装完,可以直接使用modelscope server进行模型本地启动,也可以直接启动容器,再去内部启动模型server

不带显卡的,模型本地启动命令如下:

docker run --name modelscope-server \
-p 8000:8000 registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.12.0 modelscope server --model_id=damo/ofa_ocr-recognition_handwriting_base_zh --revision=v1.0.1

带显卡的,模型本地启动命令如下:
gpus all需要安装nvidia-docker相关的,这个我的上一篇文章写了,但是**的给我屏蔽了

docker run --name modelscope-server \
--gpus all \
-p 8000:8000 registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.12.0 modelscope server --model_id=damo/ofa_ocr-recognition_handwriting_base_zh --revision=v1.0.1

不带显卡的,直接启动命令如下:

docker run --name modelscope-server \
-p 8000:8000 registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.12.0 modelscope

带显卡的,直接启动命令如下:
gpus all需要安装nvidia-docker相关的,这个我的上一篇文章写了,但是**的给我屏蔽了

docker run --name modelscope-server \
--gpus all \
-p 8000:8000 registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.12.0 modelscope

3、启动完成之后

模型本地启动的,可以直接访问地址了
文档地址:http://ip:8000/docs
获取call接口入参:http://ip:8000/describe
请求call:http://ip:8000/call
不得不说,官方写的这破返回值,我是没看懂,然后我直接看的源码,分析出来入参是什么(怎么找源码为止,看容器日志,参数请求错误会报哪个文件错了,然后直接就找到接口文件了)
接口文件地址:/opt/conda/lib/python3.10/site-packages/modelscope/server/api/routers/model_router.py
直接上入参

{
    "input": {
        "image": "http://modelscope.oss-cn-beijing.aliyuncs.com/demo/images/image_ocr_recognition.jpg"
    }
}

这个时候,即便参数对了,有的朋友还可能遇到一个问题,那就是报错!报下面这个错

AttributeError: 'dict' object has no attribute 'task_name'

这个项目真是狗,都集成进来了,源码居然还有错,经过我一点一点撸代码,发现在这个文件下【/opt/conda/lib/python3.10/site-packages/modelscope/utils/input_output.py】


这个是不对的

改成这个

这个时候再去请求一遍call方法,这次就成功了

4、容器方式直接启动,可以自己创建api服务,也可以在容器中直接modelscope server命令,但是要先把模型拉下来

4.1、自己创建api服务
cd /home
vim server.py
from flask import Flask, jsonify, request
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.outputs import OutputKeys
  
app = Flask(__name__) 
app.config['JSON_AS_ASCII'] = False  # 解决中文乱码问题

# 全局模型实例  
global_model = None 

@app.before_first_request  
def load_model():  
    global global_model  
    # 加载模型,这里替换为ModelScope的加载代码  
    # ModelScope Library >= 1.2.0
    global_model = pipeline(Tasks.ocr_recognition, model='damo/ofa_ocr-recognition_handwriting_base_zh', model_revision='v1.0.1') 

# 定义一个简单的GET接口
@app.route('/ocr', methods=['POST'])
def ocr():
    # 使用全局模型进行推理  
    global global_model 
    result = global_model(request.form.get('imageUrl'))
    return result[OutputKeys.TEXT]
  
# 启动服务器(默认端口5000,可以通过环境变量FLASK_RUN_PORT修改)  
if __name__ == '__main__':  
    app.run(host='0.0.0.0', debug=True, port=8000) # 监听所有接口
# 直接启动
python server.py
# 后台启动方式
nohup python server.py &>/home/log.txt & 
4.2、容器中直接modelscope server命令
MODELSCOPE_CACHE=/mnt/workspace/.cache/modelscope modelscope server --model_id=damo/ofa_ocr-recognition_handwriting_base_zh --revision=v1.0.1

5、至此在线搭建OCR大模型就OK了,下面讲解离线搭建OCR大模型

6、别问为什么,下面指令挨个执行

# 执行域名解析
echo '0.0.0.0 www.modelscope.cn' >> /etc/hosts
# 先看看,修改前是False
cat /opt/conda/lib/python3.10/site-packages/modelscope/hub/file_download.py|grep "local_files_only: Optional\[bool\]"
local_files_only: Optional[bool] = False,
# 执行修改
sed -i 's/local_files_only: Optional\[bool\] = False,/local_files_only: Optional\[bool\] = True,/' /opt/conda/lib/python3.10/site-packages/modelscope/hub/file_download.py
# 修改后是True
cat /opt/conda/lib/python3.10/site-packages/modelscope/hub/file_download.py|grep "local_files_only: Optional\[bool\]"
local_files_only: Optional[bool] = True,
# 修改前,先看看
cat /opt/conda/lib/python3.10/site-packages/modelscope/utils/input_output.py|grep "model=model_id,"
model=model_id,
# 执行修改
sed -i "s#model=model_id,#model=cfg.filename.replace('/configuration.json', ''),#" /opt/conda/lib/python3.10/site-packages/modelscope/utils/input_output.py
# 修改后
cat /opt/conda/lib/python3.10/site-packages/modelscope/utils/input_output.py|grep "model="
model=cfg.filename.replace('/configuration.json', ''),
# 修改前,先看看
cat /opt/conda/lib/python3.10/site-packages/modelscope/utils/pipeline_schema.json|tail -n 2
  },
}
# 执行修改
line_num=$(($(wc -l < /opt/conda/lib/python3.10/site-packages/modelscope/utils/pipeline_schema.json) - 1))
sed -i "${line_num}s/},/}/" /opt/conda/lib/python3.10/site-packages/modelscope/utils/pipeline_schema.json
# 修改后,倒数第二号逗号去了
cat /opt/conda/lib/python3.10/site-packages/modelscope/utils/pipeline_schema.json|tail -n 2
  }
}
自己写API的话,还需要改的地方,网上一点资料都没有,全靠自己撸代码!
cat /opt/conda/lib/python3.10/site-packages/modelscope/hub/snapshot_download.py|grep "local_files_only: Optional\[bool\]"
sed -i 's/local_files_only: Optional\[bool\] = False,/local_files_only: Optional\[bool\] = True,/' /opt/conda/lib/python3.10/site-packages/modelscope/hub/snapshot_download.py
# 将is_official_hub_impl方法直接返回True
vim /opt/conda/lib/python3.10/site-packages/modelscope/pipelines/util.py

再次执行命令,启动成功!

MODELSCOPE_CACHE=/mnt/workspace/.cache/modelscope modelscope server --model_id=damo/ofa_ocr-recognition_handwriting_base_zh --revision=v1.0.1
image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,772评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,458评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,610评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,640评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,657评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,590评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,962评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,631评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,870评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,611评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,704评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,386评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,969评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,944评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,179评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,742评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,440评论 2 342

推荐阅读更多精彩内容