使用 MobileNet v2 重新训练图像分类器

数据挖掘 过拟合 成立之初
2022-03-17 01:05:39

我正在使用我自己的数据集来重新训练 mobilenet_v2_100_224 模型,我目前有 4 个类,每个类有超过 100 个图像,即使我已经使用--random_scale--random_brightness参数,我仍然观察到过度拟合,如果没有这样的参数,我应该如何克服这个过度拟合问题retrain.py脚本中可用的正则化技术?以下是我的 TensorBoard 可视化:

重新训练命令:

python retrain.py --tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/feature_vector/2 --how_many_training_steps 500 --random_scale=5 --random_brightness=10 --output_graph=./retrained_graph.pb --output_labels=./retrained_labels.txt --image_dir ./data --summaries_dir /tmp/log

在此处输入图像描述

1个回答

我没有看到任何过度拟合。过度拟合是指您的验证损失随着时间的推移而变得更糟,而训练损失却有所改善。仅仅因为您的训练误差小于验证(这是预期的)并不意味着它过拟合。

但是您可以尝试冻结第一层(直到完全连接的一层)以确保网络不会失去泛化能力。