颜色分类leetcode keras imagenet 从头开始训练KerasImageNet(ILSVRC2012)图像分类模...

颜色分类leetcode keras_imagenet这个存储库包含我用来从头开始训练Keras ImageNet (ILSVRC2012)图像分类模型的代码。亮点#1 ：我使用和来加速训练管道的数据摄取。通过这种方式，我可以对数据预处理（包括在线数据增强）任务进行多重处理，并最大限度地利用GPU。亮点#2 ：除了数据增强（随机颜色失真、旋转、翻转和裁剪等），我还使用各种技巧来尝试为训练后的图像分类模型实现最佳精度。更具体地说，我为Keras模型实现了“LookAhead”优化器()、“iter_size”和“L2正则化”，并尝试使用“AdamW”（具有解耦权重衰减的亚当优化器）。亮点#3：我还开发了关于如何使用TensorRT优化经过训练的tf.keras模型的代码/文档。详情请参阅和。我从tensorflow中获取了大部分数据集准备代码。它在指定的Apache许可下。否则，请参阅以下博客文章，了解有关代码的更多实现细节：先决条件此存储库中的数据集和CNN模型是使用tf.keras ( tensorflow.kera