【发布时间】:2018-11-17 09:19:18
【问题描述】:
我有在我的工作场所使用的独特产品的图片。我无法想象初始数据库已经有类似的项目,它已经被训练过。
我尝试使用 YOLO 训练模型。这花了很长时间。时期之间可能有 7 分钟;由于数据量小,我想做 1000 个 epoch。
我在 1.0 GPU 上使用 tiny-yolov2-voc cfg/weight。我有一个项目的视频,但我把它分成框架,这样我就可以注释了。然后我尝试在图像(不是视频)上进行训练。产品与医疗保健相关。基本上医院会使用的任何东西。
我还对从 Google 获得的图像使用了 inception 方法。我注意到 inception 方法非常快并且可以产生准确的预测。但是,我担心我的图像太独特而无法正常工作。
哪种方法最好用?
如果您推荐YOLO,能否请您提供如何加快训练阶段的建议?
如果您推荐 inception,您能否解释一下为什么它适用于独特的图像?我想我很难理解 inception 如何知道我正在尝试训练哪个项目,而无需我提供注释。
提前致谢
【问题讨论】:
-
请在您的问题中添加详细信息,例如您是否安装了暗网或 COCO,以及您使用的是哪个版本的 YOLO?它在 CPU 或 GPU 上运行?你训练单个图像或视频?添加任何有用的东西。了解您的特殊产品的性质也可能会有所帮助。谢谢!
-
谢谢大卫。我现在添加了一些细节。另外,我很确定它的暗网。这是我使用的 github repo 的链接。 github.com/thtrieu/darkflow
标签: tensorflow yolo