axinc-ai / ailia-models-tflite

Quantized version of model library
25 stars 2 forks source link

ADD face_classification #52

Open kyakuno opened 1 year ago

kyakuno commented 1 year ago

https://github.com/axinc-ai/ailia-models/tree/master/face_recognition/face_classification

craft-c commented 1 year ago

現状メモ

・画像の推論結果良好 (float32モデル)

: emotion_class_count=3 : + idx=0 : category=6 [ neutral ] : prob=0.4564482271671295 : + idx=1 : category=4 [ sad ] : prob=0.21134254336357117 : + idx=2 : category=3 [ happy ] : prob=0.17144642770290375 : : [0 1] : gender_class_count=2 : + idx=0 : category=0 [ female ] : prob=0.9281189441680908 : + idx=1 : category=1 [ male ] : prob=0.07188108563423157 : Script finished successfully.