Tensorflow模型仅检测1个类别

Question

我想认出我的拳头和手掌。

所以我遵循Tensorflow Object Dection API

但是我可以看到我的模型只能识别一只手掌

我将手掌和拳头的图像数量调整为相同。

我还修改了标签文件generate_tfrecord，ssd_mobilenet_v1_pets。

我想训练模型以检测2个班级，但训练后只能识别1个班级

Answer 1

您需要检查ssd_mobilenet_v1_pets.config文件中的NUM_CLASS，并正确地匹配generate_tfrecord和labelmap.pbtxt。 满足您的需求：

NUM_CLASS = 2＃必须在配置以及用于加载训练后的模型的代码中对此进行更改。

配置必须像这样：

item {
  id: 1
  name: "fist"
}
item {
  id: 2
  name: "palm"
}

generate_tfrecord.py：

def class_text_to_int(row_label):
    if row_label == 'fist':
        return 1
    elif row_label == 'palm':
        return 2
    else:
        return None

我认为这对您来说很好。