首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >何时停止训练对象检测tensorflow

何时停止训练对象检测tensorflow
EN

Stack Overflow用户
提问于 2018-01-25 00:33:45
回答 1查看 4.5K关注 0票数 4

我正在使用google api(faster_rcnn_inception_resnet_v2_atrous_coco).中提供的预训练模型在水果数据集上训练更快的rcnn模型。

我对默认配置做了很少的更改。(类数: 12 fine_tune_checkpoint:预训练检查点模型的路径,from_detection_checkpoint: true)。我拥有的带注释的图像总数约为12000张。

经过9000个步骤的训练后,我得到的结果的准确率低于1,尽管我希望它至少是50% (在评估中没有检测到任何东西,因为准确率几乎为0)。损失在0到4之间波动。

我应该训练它的步数是多少。我读过一篇文章,说跑大约80万步,但这是你从零开始训练的步数?

由于类的数量不同,模型的FC层发生了变化,但它不应该影响那些已经存在于预训练模型中的类,比如“apple”?

任何帮助都将不胜感激!

EN

回答 1

Stack Overflow用户

发布于 2018-01-27 06:43:15

你不应该看你的训练损失来决定什么时候停止。相反,您应该定期通过评估器运行模型,并在评估mAP停止改进时停止训练。

票数 5
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/48427381

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档