安装MindSpore Vision

环境依赖

• numpy 1.17+

• opencv-python 4.1+

• pytest 4.3+

• MindSpore 1.5+

• ml_collection

• tqdm

• pillow

安装

环境准备

• 创建一个conda虚拟环境并且激活。

  conda create -n mindvision python=3.7.5 -y
  conda activate mindvision
  

• 安装MindSpore

  pip install mindspore
  

安装MindSpore Vision

• 源码安装

  git clone https://gitee.com/mindspore/vision.git
  cd vision
  python setup.py install
  

• 使用pip安装

  pip install mindvision
  

验证

为了验证MindVision和所需的环境是否正确安装，我们可以运行示例代码来初始化一个分类器然后推理一张图片。

推理所用的图片 来自MNIST数据集，用户可以通过参数device_target来自定义推理所用的平台。

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

{4: 'four'}

如果您成功安装，以上代码应该会成功运行。