超级会员免费看
| 自定义数据集(宝可精灵数据集)实现图像分类+补充:tf.where！ |
文章目录
一. 数据集介绍以及加载
1.1. 数据集简单描述
- 我们收集了宝可精灵(动漫)视频片段，从中收集了
5种精灵，每个精灵有各种形态的图片。其中：皮卡丘234张图片，超梦239张，杰尼龟223张，小火龙238张，妙蛙种子234张。
- 数据集划分： 每一类别的所有图片，按照下面这个比例进行划分。其实这个比例并不是针对每个类别进行提取
60%(600多张)，而是针对总体的1168张图片。
本文介绍了使用TensorFlow2.0处理自定义的宝可梦图像数据集进行分类的方法，包括数据集加载、预处理、模型构建、训练与优化。通过数据增强、迁移学习等手段解决小样本训练问题，探讨了tf.where的使用，并分享了进一步改进的技巧。
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
