上传者: hhhhhhhhhhwwwwwwwwww
|
上传时间: 2022-01-23 09:15:29
|
文件大小: 936.29MB
|
文件类型: ZIP
本例提取了猫狗大战数据集中的部分数据做数据集,演示tensorflow2.X版本如何使用Keras实现图像分类,分类的模型使用ResNet50。本文实现的算法有一下几个特点:
1、自定义了图片加载方式,更加灵活高效,不用将图片一次性加载到内存中,节省内存,适合大规模数据集。
2、加载模型的预训练权重,训练时间更短。
3、数据增强选用albumentations。