VSCode一键接入Notebook体验算法套件快速完成水表读数

2023-02-09 00:00:00 修改 数据 模型 算法 推理

本文分享自华为云社区《VSCode一键接入Notebook体验算法套件快速完成水表读数》，作者：HWCloudAI。

本示例围绕真实AI需求场景，介绍VSCode一键接入Notebook体验算法套件快速完成水表读数的使用流程。

算法开发套件中目前提供自研(ivg系列)和开源(mm系列)共两套算法资产，可应用于分类、检测、分割和OCR等任务中。本示例中将组合使用自研分割算法(ivgSegmentation)和开源OCR算法(mmOCR)完成水表读数识别项目，并使用算法开发套件将其部署为华为云在线服务。

说明：

本案例教程仅适用于“华北-北京四”区域，新版Notebook。

准备数据

  1. 登录OBS控制台，创建OBS对象桶，区域选择“华北-北京四”。
  2. 登录ModelArts控制台，选择控制台区域为“华北-北京四”。
  3. 在“全局配置”页面查看是否已经配置授权，允许ModelArts访问OBS。如果没有配置授权，请参考配置访问授权（全局配置）添加授权。
  4. 分别下载本案例的数据集，水表表盘分割数据集和水表表盘读数OCR识别数据集到OBS桶中，OBS路径示例如下

obs://{OBS桶名称}/water_meter_segmentation 水表表盘分割数据集

obs://{OBS桶名称}/water_meter_crop 水表表盘读数OCR识别数据集

说明：

从AIGallery下载数据集免费，但是数据集存储在OBS桶中会收取少量费用，具体计费请参见OBS价格详情页，案例使用完成后请及时清除资源和数据。

准备开发环境

在“ModelArts控制台 > 开发环境 > Notebook（New）”页面中，创建基于pytorch1.4-cuda10.1-cudnn7-ubuntu18.04镜像，类型为GPU的Notebook，具体操作请参见创建Notebook实例章节。

本案例需要使用VS Code 远程连接Notebook，需要开启SSH远程开发。

图1 创建Notebook实例

1.实例的密钥文件需要下载至本地的如下目录或其子目录中：

Windows：C:\Users{{user}}

Mac/Linux: Users/{{user}}

2.在ModelArts控制台->开发环境 Notebook，单击“操作”列的“更多 > VS Code接入”。

如果本地已安装VS Code，请单击“打开”,进入“Visual Studio Code”页面。

如果本地未安装VS Code，请根据实际选择“win”或“其他”下载并安装VS Code。VS Code安装请参考安装VS Code软件。

如果用户之前未安装过ModelArts VS Code插件，此时会弹出安装提示，请单击“Install and Open”进行安装；如果之前已经安装过插件，则不会有该提示，请跳过此步骤，直接执行后面步骤

安装过程预计1~2分钟，安装完成后右下角会弹出对话框，请单击“Reload Window and Open”。

在弹出的提示中，勾选“Don’t ask again for this extension”，然后单击"Open"。

3.远程连接Notebook实例。

  • 远程连接执行前，会自动在（Windows：C:\Users{{user}}.ssh或者downloads，Mac/Linux: Users/{{user}}/.ssh或者downloads）目录下根据密钥名称查找密钥文件，如果找到则直接使用该密钥打开新窗口并尝试连接远程实例，此时无需选择密钥。
  • 如果未找到会弹出选择框，请根据提示选择正确的密钥。
  • 如果密钥选择错误，则弹出提示信息，请根据提示信息选择正确密钥。
  • 当弹出提醒实例连接失败，请关闭弹窗，并查看OUTPUT窗口的输出日志，请查看FAQ并排查失败原因。

使用算法套件进行开发

Step1 创建算法工程

1.成功接入之后，在VS Code页面点击文件->打开文件夹，选择如下文件夹打开

2.新建终端

3.在work目录下执行

ma-cli createproject

相关文章