单个对象检测keras

Question

我想创建一个使用keras识别单个对象的系统。在我的情况下，我将检测车轮。如何仅为1个对象训练我的系统？我在使用猫和狗之前做了分类任务，但现在它完全不同了。我仍然“分类”，0级=轮子，等级= 1非轮（只是随机图像的任何东西）？如何在此问题中执行以下步骤？ 1）1个物体的训练系统 2）检测对象（滑动窗口或热图）

Answer 1

您的任务是所谓的二进制分类。 确保你的最后一层只有一个神经元（例如对于Sequential model model.add（Dense（1，... other parameters ...）））并使用binary_crossentropy作为损失函数。

希望这会有所帮助。