一、准备工作
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