我有将图像分类为“裸”或“非裸”的代码。它是通过张量流python在深度学习上实现的。可以在Tensorflow Implementation of Yahoo's Open NSFW Model
中建立代码我想在数据集中添加更多图像以便进行微调。如何通过使用另一个数据集在此实现中进行微调。
答案 0 :(得分:1)
只需加载其model并使用ones they provide初始化其权重,就像他们如何here一样。假设您熟悉张量流,则应继续在图像上训练该模型。
除了this blog post外,我不知道该团队在他们的工作上发表过任何其他出版物。这是一个问题,因为他们没有说明他们的训练参数(优化器的选择,学习率等)。如果要微调此模型,则必须在这方面进行一些试验。
答案 1 :(得分:0)
它们是否为您提供了训练提供的模型所依据的原始数据集?如果是这样,您可以轻松地将自己的数据集添加到他们的数据集中,并根据组合的数据集训练一个全新的模型。
我写了更多有关此“组合”数据集的内容,您可以在其中添加更多或更少的数据here。
祝你好运!