颜色分类leetcode keras imagenet: 从头开始训练Keras ImageNet(ILSVRC2012)图像分类模型

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