带你体验mindcv (一)

简介

MindCV是一个基于 MindSpore 开发的,致力于计算机视觉相关技术研发的开源工具箱。

它提供大量的计算机视觉领域的经典模型和SoTA模型以及它们的预训练权重和训练策略。

其中模型包含了VGG,ResNet,MobileNet,googlenet等常用的CV网络。

每个模型的配置文件介绍里面都有相关模型的精度数据。还提供了预训练权重的下载地址。

预训练权重的好处不用多说,微调下就能直接使用,省去了大量的训练时间。

同时,还提供了自动增强等SoTA算法来提高模型性能。通过解耦的模块设计,您可以轻松地将MindCV应用到您自己的CV任务中。

mindcv 开源地址 mindcv: MindSpore Computer Vision is an open source computer vision research toolbox based on MindSpore in computer vision direction.

如果是使用cifar10 数据集,一行代码就能把创建一个数据集对象,并把数据集下载好。

不过当前仅支持Mnist cifar10 和cifar100, 希望后续能增加更多的数据集支持。

如下代码可以看出当前下载支持的数据集。

mindcv/data/dataset_download.py
__all__ = [
    "get_dataset_download_root",
    "MnistDownload",
    "Cifar10Download",
    "Cifar100Download",
]

复制

创建网络也很方便,设置pretrained=True 就能直接下载预训练权重,真的很方便。

cke_138928.png

甚至还提供了模型名查询的接口。

提供了训练,验证,推理的脚本。

比如拿推理来说。

推理的图片如下:

cke_422100.png

结果如下,这个就是开箱即用啊。

训练也是如此。