14、MiniGoogLeNet在CIFAR - 10和Tiny ImageNet上的实践

最新推荐文章于 2025-11-16 13:15:34 发布

职场萌新987

最新推荐文章于 2025-11-16 13:15:34 发布

阅读量14

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习计算机视觉实战文章标签： MiniGoogLeNet CIFAR-10 Tiny ImageNet

本文链接：https://blog.youkuaiyun.com/go5gopher/article/details/154816993

深度学习计算机视觉实战专栏收录该内容

26 篇文章 ¥499.90

订阅专栏¥69.90

会员秒杀 ¥9.9 重磅福利

超级会员免费看

MiniGoogLeNet在CIFAR - 10和Tiny ImageNet上的实践

1. MiniGoogLeNet在CIFAR - 10上的训练

在使用MiniGoogLeNet对CIFAR - 10数据集进行训练时，我们需要进行一系列的准备工作。首先，构建数据增强器：

aug = ImageDataGenerator(rotation_range=18, zoom_range=0.15,
                         width_shift_range=0.2, height_shift_range=0.2,
                         shear_range=0.15, horizontal_flip=True,
                         fill_mode="nearest")

接着，构建回调函数集来监控训练进度并调用学习率调度器：

figPath = os.path.sep.join([args["output"], "{}.png".format(os.getpid())])
jsonPath = os.path.sep.join([args["output"], "{}.json".format(os.getpid())])
callbacks = [TrainingMonitor(figPath, jsonPath=jsonPath),
             LearningRateScheduler(poly_decay)]