这是CSDN博客上面的一篇教程 地址是：https://blog.csdn.net/alxws/article/details/140059294

运行步骤： 准备数据集，将一类图片，如imagenet_train中的某个文件夹中的所有图片直接放入到dataset文件夹中，中间不要有其他文件夹。 运行python train.py，训练结束之后会产生一个模型文件。 然后运行test.py，完成之后，在results文件夹中会出现生成的图片。