您好,欢迎访问代理记账网站
  • 价格透明
  • 信息保密
  • 进度掌控
  • 售后无忧

实验四:迁移学习finetune:加载Resnet预训练的参数,修改Resnet的fc层,重新训练,完成4分类任务

一、准备工作
1. 模型准备

    模型下载:https://download.pytorch.org/models/resnet18-5c106cde.pth

    下载完后如下,有44.6M。

    

2. 数据集准备

   |----data

            |----pubu                                       #下载的数据集。

                       |----train

                                |----saopin                #136张图片。

                                |----wurenji               #136张图片。

                       |----val

                                |----saopin                #70张图片。

                                |----wurenji               #70张图片。

            |----resnet18-5c106cde.pth        #预训练的模型

   |----src

            |-----finetune_resnet18.py          #要写的代码

   |----tools                                              #通用的一些函数

            |----my_dataset.py                      #Dataset
 

 

 

 

 

 


分享:

低价透明

统一报价,无隐形消费

金牌服务

一对一专属顾问7*24小时金牌服务

信息保密

个人信息安全有保障

售后无忧

服务出问题客服经理全程跟进