MindStudio训练营第一季-高阶班大作业
获取源码
下载ReID源码
ATC ReID (FP16)模型地址:ATC ReID (FP16)-昇腾社区 (hiascend.com)
点击下载获取下载链接,在ecs服务器上下载代码
wget https://ascend-repo-modelzoo.obs.cn-east-2.myhuaweicloud.com/model/ATC%20ReID%28FP16%29%20%20from%20Pytorch%20-%20Ascend310/zh/1.1/ATC%20ReID%28FP16%29%20%20from%20Pytorch%20-%20Ascend310.zip --no-check-certificate
解压代码
进入ReID_for_Pytorch目录里下载ReID原仓库代码
当前目录结构如下, pth模型文件已经在下载的压缩包中
模型使用Market1501数据集的测试集,自行下载后请上传到服务器自定义目录
Mindstudio工具操作
安装和配置环境
按照Mindstudio用户手册中的安装指南—>安装操作来安装Mindstudio。
推理环境在远端服务器上已经配置完成(系统提供的镜像已经设置好环境)。
创建工程
本机windows打开Mindstudio,选择file-new-project新建
选择Ascend App 项目,选择ACL Project(python)项目
设置项目名称和本地路径
配置项目结构
选择add python SDK
选择ssh interpreter
配置本地文件夹与远程服务器映射
拉取代码
成功拉取代码
数据预处理
数据预处理将原始数据集转换为模型输入的数据,使用ReID_preprocess.py(找到该文件,然后右键选择Modify Run Configuration)
gallery同理也是同样的操作
第一个参数为生成的数据集文件格式,第二个参数为预处理后的数据文件的相对路径,第三个参数为生成的数据集文件保存的路径。运行成功后,在当前目录中生成prep_bin.info。
运行ReID_pth2onnx文件将pth文件导出为onnx文件
模型转换:onnx 转om(点击AscendModel Converter)
专家工具体验
单击菜单栏“Ascend > Advisor > Advisor”,弹出专家系统工具界面
左上角“New Project”按钮,打开专家系统配置界面,可选择Remote Run和Local Run模式且每种模式下可选择OM only和Inference App专家系统分析模式
- 点赞
- 收藏
- 关注作者
评论(0)