单目标检测 keras

Question

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

Answer 1

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

希望这会有所帮助。