【发布时间】:2018-06-23 10:16:33
【问题描述】:
我有将图像分类为裸体或非裸体的代码。它是使用张量流 python 在深度学习上实现的。代码可以建立在Tensorflow Implementation of Yahoo's Open NSFW Model
我想在数据集中添加更多图像以进行微调。如何使用另一个数据集在此实现中进行微调。
【问题讨论】:
标签: python tensorflow neural-network deep-learning
我有将图像分类为裸体或非裸体的代码。它是使用张量流 python 在深度学习上实现的。代码可以建立在Tensorflow Implementation of Yahoo's Open NSFW Model
我想在数据集中添加更多图像以进行微调。如何使用另一个数据集在此实现中进行微调。
【问题讨论】:
标签: python tensorflow neural-network deep-learning
只需加载他们的model 并使用ones they provide 初始化其权重,类似于他们的操作here。假设您熟悉 tensorflow,那么您应该继续在图像上训练该模型。
除了this blog post,我不知道该团队还发表了关于他们工作的任何其他出版物。这有点问题,因为他们没有说明他们的训练参数(优化器的选择、学习率等)。如果你想微调这个模型,你将不得不在这方面进行一些实验。
【讨论】:
他们是否为您提供了用于训练所提供模型的原始数据集?如果是这样,您可以轻松地将自己的数据集添加到他们的数据集中,并基于组合数据集训练一个全新的模型。
我写了更多关于这个“组合”数据集的文章,您可以在其中添加更多或更少的数据,here。
祝你好运!
【讨论】: