如何使用自定义数据集更有效地训练vgg16模型

2024-04-27 02:36:20 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经从零开始实现了一个带有所有层的VGG16模型。当我用我的自定义图像数据集训练它时,完成一个历元需要5个小时,而且很耗时。我就在那儿停了下来。你知道吗

因为我的数据集非常独特,所以如何更快地训练我的模型。有没有一种方法可以简化我的训练任务,这样我就可以获得很好的准确性。你知道吗

我的数据集有5个培训班,每个班有3000个彩色图像,每个班有1000个彩色图像,分别放在两个单独的验证和培训文件夹中进行验证。我正在使用Keras来构建我的VGG16模型。你知道吗


Tags: 数据方法模型图像文件夹彩色图像keras小时