MindSpore Vision—易用易理解的主流工具库
MindSpore Vision是基于昇思MindSpore的开源计算机视觉研究工具库,MindSpore Vision已构筑基础的AI复用能力,支持部分主流、前沿网络,例如Efficientnet、Lenet、Mobilenetv2、Resnet、Vit等,提供手机侧推理与部署,欢迎开发者们参与贡献,共同拓展和维护MindSpore Vision,力争将其打造为易用易理解的主流工具库。
详情参考:MindSpore Vision 文档 — MindSpore master documentation
1.安装
pip install mindvision
2.验证安装
下载源码,源码里面有example
windows下肯定要修改下命令
python ./examples/classification/lenet/lenet_mnist_infer.py --data_url ./tests/st/classification/dataset/mnist/mnist.jpg --pretrained True
报错,看下代码
默认GPU,加参数 --device_target CPU
又错了,报文件不存在
什么,代码里面的lenet的url为空
再看下最新master代码
pip安装的不是最新的代码,手动修改下
然后就可以了