You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

pipeline.md 3.3 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. # Pipeline使用教程
  2. 本文将简单介绍如何使用`pipeline`函数加载模型进行推理。`pipeline`函数支持按照任务类型、模型名称从模型仓库
  3. 拉取模型进行进行推理,当前支持的任务有
  4. * 人像抠图 (image-matting)
  5. * 基于bert的语义情感分析 (bert-sentiment-analysis)
  6. 本文将从如下方面进行讲解如何使用Pipeline模块:
  7. * 使用pipeline()函数进行推理
  8. * 指定特定预处理、特定模型进行推理
  9. * 不同场景推理任务示例
  10. ## 环境准备
  11. 详细步骤可以参考 [快速开始](../quick_start.md)
  12. ## Pipeline基本用法
  13. 1. pipeline函数支持指定特定任务名称,加载任务默认模型,创建对应Pipeline对象
  14. 执行如下python代码
  15. ```python
  16. >>> from modelscope.pipelines import pipeline
  17. >>> img_matting = pipeline(task='image-matting', model='damo/image-matting-person')
  18. ```
  19. 2. 传入单张图像url进行处理
  20. ``` python
  21. >>> import cv2
  22. >>> result = img_matting('http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/data/test/maas/image_matting/test.png')
  23. >>> cv2.imwrite('result.png', result['output_png'])
  24. >>> import os.path as osp
  25. >>> print(f'result file path is {osp.abspath("result.png")}')
  26. ```
  27. pipeline对象也支持传入一个列表输入,返回对应输出列表,每个元素对应输入样本的返回结果
  28. ```python
  29. >>> results = img_matting(
  30. [
  31. 'http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/data/test/maas/image_matting/test.png',
  32. 'http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/data/test/maas/image_matting/test.png',
  33. 'http://pai-vision-data-hz.oss-cn-zhangjiakou.aliyuncs.com/data/test/maas/image_matting/test.png',
  34. ])
  35. ```
  36. 如果pipeline对应有一些后处理参数,也支持通过调用时候传入.
  37. ```python
  38. >>> pipe = pipeline(task_name)
  39. >>> result = pipe(input, post_process_args)
  40. ```
  41. ## 指定预处理、模型进行推理
  42. pipeline函数支持传入实例化的预处理对象、模型对象,从而支持用户在推理过程中定制化预处理、模型。
  43. 下面以文本情感分类为例进行介绍。
  44. 由于demo模型为EasyNLP提供的模型,首先,安装EasyNLP
  45. ```shell
  46. pip install https://atp-modelzoo-sh.oss-cn-shanghai.aliyuncs.com/release/package/whl/easynlp-0.0.4-py2.py3-none-any.whl
  47. ```
  48. 下载模型文件
  49. ```shell
  50. wget https://atp-modelzoo-sh.oss-cn-shanghai.aliyuncs.com/release/easynlp_modelzoo/alibaba-pai/bert-base-sst2.zip && unzip bert-base-sst2.zip
  51. ```
  52. 创建tokenizer和模型
  53. ```python
  54. >>> from modelscope.models import Model
  55. >>> from modelscope.preprocessors import SequenceClassificationPreprocessor
  56. >>> model = Model.from_pretrained('damo/bert-base-sst2')
  57. >>> tokenizer = SequenceClassificationPreprocessor(
  58. model.model_dir, first_sequence='sentence', second_sequence=None)
  59. ```
  60. 使用tokenizer和模型对象创建pipeline
  61. ```python
  62. >>> from modelscope.pipelines import pipeline
  63. >>> semantic_cls = pipeline('text-classification', model=model, preprocessor=tokenizer)
  64. >>> semantic_cls("Hello world!")
  65. ```
  66. ## 不同场景任务推理示例
  67. 人像抠图、语义分类建上述两个例子。 其他例子未来添加。