Image classification method and device, storage medium and terminal

The invention discloses an image classification method and device, a storage medium and a terminal. The method comprises the steps of obtaining a to-be-classified target image; inputting the to-be-classified target image into a pre-trained image classification model, and outputting a plurality of ca...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: QI XIAODONG, SHI GANGJIE, LIAO DANPING, WANG GUOLONG
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:The invention discloses an image classification method and device, a storage medium and a terminal. The method comprises the steps of obtaining a to-be-classified target image; inputting the to-be-classified target image into a pre-trained image classification model, and outputting a plurality of category scores corresponding to the to-be-classified target image; the loss function of the pre-trained image classification model comprises a cross entropy loss function and model length constraint loss of image classification; and determining the category corresponding to the maximum score in the plurality of category scores as the final category of the target image to be classified. The corresponding model length constraint loss is added into the loss function of the model, the loss can enable the model to be quickly corrected from the wrong classification category to the correct classification category in the training stage, and after the classification is performed to the correct category, the classification co