Method for detecting defect and method for training model

Abstract

The present disclosure provides a method and device for detecting an image category. The method includes: acquiring a sample data set including a plurality of sample images labeled with a category, the sample data set including a training data set and a verification data set; training a deep learning model using the training data set to obtain, according to different numbers of training rounds, at least two trained models; testing the at least two trained models using the verification data set to generate a verification test result; generating, based on the verification test result, a verification test index; determining, according to the verification test index, a target model from the at least two trained models; and predict a to-be-tested image of the target object using the target model to obtain the category of the to-be-tested image.