Маска RCNN предсказывает только один тип класса из нескольких классов

Question

У меня есть 8 классов в наборе рукописного текста и символов. но он предсказывает только один тип из 8 классов. Я использую Математическая реализация .

class SymbolConfig(Config):

# Give the configuration a recognizable name
NAME = "symbols"

IMAGES_PER_GPU = 2

# Number of classes (including background)
NUM_CLASSES = 1 + 7  # Background + 

# Number of training steps per epoch
STEPS_PER_EPOCH = 100

# Skip detections with < 90% confidence
DETECTION_MIN_CONFIDENCE = 0.9

Answer 1

Поскольку вам также необходимо добавить классы в функцию load_custom ().

например:

self.add_class("datasetName", 1, "class1")
self.add_class("datasetName", 2, "class2")
......
self.add_class("datasetName", 8, "class8")