【问题标题】:Fine tuning a deep learning model using my own dataset使用我自己的数据集微调深度学习模型
【发布时间】:2018-06-23 10:16:33
【问题描述】:

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

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

【问题讨论】:

    标签: python tensorflow neural-network deep-learning


    【解决方案1】:

    只需加载他们的model 并使用ones they provide 初始化其权重,类似于他们的操作here。假设您熟悉 tensorflow,那么您应该继续在图像上训练该模型。

    除了this blog post,我不知道该团队还发表了关于他们工作的任何其他出版物。这有点问题,因为他们没有说明他们的训练参数(优化器的选择、学习率等)。如果你想微调这个模型,你将不得不在这方面进行一些实验。

    【讨论】:

      【解决方案2】:

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

      我写了更多关于这个“组合”数据集的文章,您可以在其中添加更多或更少的数据,here

      祝你好运!

      【讨论】:

        猜你喜欢
        • 2016-11-24
        • 1970-01-01
        • 2018-03-29
        • 1970-01-01
        • 2018-10-27
        • 2023-04-10
        • 2023-03-29
        • 1970-01-01
        • 2020-06-02
        相关资源
        最近更新 更多