安装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'}

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