MindSpore Vision—易用易理解的主流工具库
MindSpore Vision是基于昇思MindSpore的开源计算机视觉研究工具库,MindSpore Vision已构筑基础的AI复用能力,支持部分主流、前沿网络,例如Efficientnet、Lenet、Mobilenetv2、Resnet、Vit等,提供手机侧推理与部署,欢迎开发者们参与贡献,共同拓展和维护MindSpore Vision,力争将其打造为易用易理解的主流工具库。

详情参考:https://www.mindspore.cn/visi...

image.png

1.安装

pip install mindvision

2.验证安装

下载源码,源码里面有example

https://gitee.com/mindspore/v...

image.png

windows下肯定要修改下命令

python ./examples/classification/lenet/lenet_mnist_infer.py --data_url ./tests/st/classification/dataset/mnist/mnist.jpg --pretrained True

image.png

报错,看下代码
image.png

默认GPU,加参数 --device_target CPU

image.png

又错了,报文件不存在

image.png

什么,代码里面的lenet的url为空

再看下最新master代码

image.png

pip安装的不是最新的代码,手动修改下

image.png

然后就可以了

MindSpore Vision下载数据集很方便,还能下载主流的模型,很方便。

image.png

邮箱:8714604@qq.com


sktier
1 声望0 粉丝