Abstract: Embodiments disclosed herein relate to a method, a device, and a computer program product for training an image classification model. The method includes: determining a first classification result obtained by the image classification model on pixel points of a sample image in a first training iteration, wherein the first classification result indicates that each of the pixel points belongs to one of a plurality of object classes; determining a second classification result obtained by the image classification model on the pixel points of the sample image in a second training iteration; determining, based on the first classification result and the second classification result, a first set of recall rates and a second set of recall rates for the plurality of object classes; and adjusting, based on the first set of recall rates and the second set of recall rates, the image classification model to obtain a trained image classification model.
Type:
Grant
Filed:
June 27, 2022
Date of Patent:
January 14, 2025
Assignee:
Dell Products L.P.
Inventors:
Bin He, Zijia Wang, Zhen Jia, Jinpeng Liu