Training device, object detection device, training method, object detection method, and non-transitory computer-readable storage medium

Abstract

A training device acquires a degree of similarity between a correct answer region indicating a region of an object in an image and each of a plurality of anchor boxes set in advance in an image, selects, among the plurality of anchor boxes, an anchor box for which the degree of similarity is greater than or equal to a predetermined threshold, for the correct answer region, and performs training of a neural network model for detecting the object, based on the correct answer region and the selected anchor box. If a maximum number of anchor boxes have been selected for the correct answer region, the training device changes the maximum number of the anchor boxes for the correct answer region based on the degrees of similarity acquired for the anchor boxes selected for the correct answer region.