Electronic device for augmenting training data, and control method therefor

An electronic device comprising:

• one or more processors;
• and
• memory, storing:
• a first training data set comprising a plurality of pieces of 2D pose data and a plurality of pieces of 3D pose data corresponding to the plurality of pieces of 2D pose data;
• and
• instructions that, when executed by the one or more processors, cause the electronic device to:
• train a first neural network model to estimate 3D poses based on the first training data set;
• obtain an augmented data set by augmenting the first training data set,
• based on at least one of similarity or reliability of 3D pose augmented data included in the augmented data set;
• select at least one piece of 3D pose augmented data among a first plurality of pieces of 3D pose augmented data included in the augmented data set;
• obtain a second training data set comprising the 3D pose augmented data and 2D pose augmented data corresponding to the 3D pose augmented data;
• and
• retrain the first neural network model based on the second training data set.