昇思MindSpore打卡营 9号 -ResNet50迁移学习 学习心得

下载数据集

dataset_url = "https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/notebook/datasets/intermediate/Canidae_data.zip"

通过此行代码指定下载来源url之后

download(dataset_url, "./datasets-Canidae", kind="zip", replace=True)

可通过此代码将下载数据集，并将数据集解压到当前目录下的datasets-Canidae中，便于使用

加载数据集

可以使用mindspore.dataset.ImageFolderDataset接口加载数据集，

接着用 

data_set = data_set.map(
        operations=trans,
        input_columns='image',
        num_parallel_workers=workers)

进行数据映射

训练模型

训练模型的时候，可以使用固定特征进行训练，为了不在反向传播中计算梯度，可以设置 requires_grad == False 冻结参数。

评估

预训练后训练模型，相比于未进行过预训练直接训练模型，速度提升明显