Object detection device, learning method, and recording medium
摘要:
In the object detection device, the plurality of object detection units output a score indicating a probability that a predetermined object exists, for each partial region set with respect to image data inputted. The weight computation unit computes a weight for each of the plurality of object detection units by using weight computation parameters and based on the image data. The weights are used when the scores outputted by the plurality of object detection units are merged. The weight redistribution unit changes the weight for a predetermined object detection unit, among the weights computed by the weight computation unit, to 0 and output the weights. The merging unit merges the scores outputted by the plurality of object detection units for each of the partial regions, by using the weights computed by the weight computation unit and including the weight changed by the weight redistribution unit. The loss computation unit computes a difference between a ground truth label of the image data and the merged score merged by the merging unit as a loss. Then, the parameter correction unit corrects the weight computation parameters so as to reduce the loss.
信息查询
0/0