Method, device, and computer program product for data augmentation

Abstract

Embodiments of the present disclosure relate to a method, an electronic device, and a computer program product for data augmentation. The method includes: generating a group of candidate images based on a target image by using a thermodynamic genetic algorithm (TDGA) model, the TDGA model being configured to apply one or more operations of a set of predetermined image processing operations during each evolution process; and determining multiple augmented images from the group of candidate images based on free energy of the group of candidate images, the multiple augmented images being determined as belonging to the same classification with the target image. In this way, data augmentation can be efficiently implemented by a thermodynamic genetic algorithm.