Что мне делать, если я хочу покрыть только человеческие объекты в маске R-CNN? - PullRequest
Новая
       26

Что мне делать, если я хочу покрыть только человеческие объекты в маске R-CNN?

1 голос
/ 08 ноября 2019

В настоящее время я использую маску R-CNN. Я хочу распознавать только человеческие объекты, а затем покрывать только человеческие объекты.

Поэтому я попытался стереть 80 объектов, которые были изучены в кокосовом файле, но они не работали правильно. Я пытался использовать индексный номер объекта-человека, но он не работал. 

class_names = [
    'BG', 'person',
    'bicycle', 'car', 'motorcycle', 'airplane',
    'bus', 'train', 'truck', 'boat', 'traffic light',
    'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird',
    'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear',
    'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie',
    'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball',
    'kite', 'baseball bat', 'baseball glove', 'skateboard',
    'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup',
    'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
    'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza',
    'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed',
    'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote',
    'keyboard', 'cell phone', 'microwave', 'oven', 'toaster',
    'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors',
    'teddy bear', 'hair drier', 'toothbrush'
]


 def apply_mask(image, mask, color, alpha=1):    #alpha값 0.5->1
    """apply mask to image"""
    for n, c in enumerate(color):
          image[:, :, n] = np.where(
                mask == 1,
               image[:, :, n] *(1 - alpha) + alpha * c,
               image[:, :, n]
            )

    return image

Даже если я изменяю код, программа продолжает распознавать объекты, которые не являются объектами-людьми.

...