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...
Gespeichert in:
Hauptverfasser: | , , , |
---|---|
Format: | Patent |
Sprache: | chi ; eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
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 |
---|