Training device, prediction device, training method, and recording medium

Abstract

The partial image generation means generates partial images from the input image, and the partial image selection means selects a plurality of training partial images to be used as training data from all the generated partial images. The training means trains a prediction model for predicting a probability that a predetermined feature is included in the selected training partial images. The prediction means performs prediction for all the partial images using the trained prediction model. Then, the partial image selection means selects the plurality of training partial images to be used in a next training based on predicted values for all the partial images. The training of the prediction model by the training means is repeated while the training partial images are updated.