International Business Machines Corporation
Dataset driven custom learning for multi-scale object detection

Abstract:

According to one embodiment, a method, computer system, and computer program product for generating anchors and selecting feature maps for a multi-scale object detection program based on analysis of the dataset is provided. The present invention may include generating a scale distribution of one or more scales of ground-truth objects, and, based on the scale distribution, dividing the effective scale range into a number of anchors greater than zero; furthermore, the invention may include generating a ratio distribution of ratios of the ground-truth objects; based on the ratio distribution, generating a ratio for at least one of the number of anchors; determining a template scale of one or more feature maps; and assigning the number of anchors to the feature maps based on the relative size of a scale of an anchor matching the relative size of a template scale of a feature map.