使用我自己的数据集微调深度学习模型

时间:2018-06-21 11:31:15

标签: python tensorflow neural-network deep-learning

我有将图像分类为“裸”或“非裸”的代码。它是通过张量流python在深度学习上实现的。可以在Tensorflow Implementation of Yahoo's Open NSFW Model

中建立代码

我想在数据集中添加更多图像以便进行微调。如何通过使用另一个数据集在此实现中进行微调。

2 个答案:

答案 0 :(得分:1)

只需加载其model并使用ones they provide初始化其权重,就像他们如何here一样。假设您熟悉张量流,则应继续在图像上训练该模型。

除了this blog post外,我不知道该团队在他们的工作上发表过任何其他出版物。这是一个问题,因为他们没有说明他们的训练参数(优化器的选择,学习率等)。如果要微调此模型,则必须在这方面进行一些试验。

答案 1 :(得分:0)

它们是否为您提供了训练提供的模型所依据的原始数据集?如果是这样,您可以轻松地将自己的数据集添加到他们的数据集中,并根据组合的数据集训练一个全新的模型。

我写了更多有关此“组合”数据集的内容,您可以在其中添加更多或更少的数据here

祝你好运!