search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.04k stars 325 forks source link

Add MiVOLO age estimation #1359

Closed mucunwuxian closed 9 months ago

mucunwuxian commented 10 months ago

https://github.com/axinc-ai/ailia-models/issues/66

mucunwuxian commented 10 months ago

@kyakuno こちら、すいません。🙇‍♂ その後、opset11でのonnxファイル作成が行えた為、確認を少し待って頂けたらと思います。 ailia.netでのテストを実施後、PRコードを更新させて頂きます。

mucunwuxian commented 10 months ago

opset11で作成したonnxファイルにて、ailia.net形式での処理実施に成功しました そちらの版でPRを出させて頂いています。

kyakuno commented 10 months ago

モデルをアップロードしました。 https://storage.googleapis.com/ailia-models/mivolo/mivolo.onnx など

kyakuno commented 10 months ago

MPS（FP16）だと精度が出ず、FemaleがMaleになってしまうようなので、MPSは無効化した方が良さそうですね。

    # disable FP16
    if "FP16" in ailia.get_environment(args.env_id).props or sys.platform == 'Darwin':
        logger.warning('This model do not work on FP16. So use CPU mode.')
        args.env_id = 0
kyakuno commented 10 months ago

BLASだといい感じに動きました。

スクリーンショット 2024-01-08 22 27 59

kyakuno commented 10 months ago

@mucunwuxian モデルのカテゴリにつきまして、顔以外も使っていますが、「age_estimation」から、「face_recognition」に移動いただければと考えています。

kyakuno commented 10 months ago

@mucunwuxian 後、MiVOLOを解説するBLOGを作成いただくことは可能でしょうか？MiVOLOは精度が高くて有用そうなので、技術の詳細をまとめていただけると、とても嬉しいです。

mucunwuxian commented 9 months ago

@mucunwuxian モデルのカテゴリにつきまして、顔以外も使っていますが、「age_estimation」から、「face_recognition」に移動いただければと考えています。

@kyakuno こちら、了解しました！ 対応させて頂きます。

mucunwuxian commented 9 months ago

@mucunwuxian 後、MiVOLOを解説するBLOGを作成いただくことは可能でしょうか？MiVOLOは精度が高くて有用そうなので、技術の詳細をまとめていただけると、とても嬉しいです。

@kyakuno ありがとうございます！ BLOGの作成、勿論可能です。 コードの修正が一区切り付いたところで、作成を開始させて頂きます。🙇‍♂

mucunwuxian commented 9 months ago

@kyakuno 記載頂いた事項の対応をさせて頂きました。 ご確認の程、何卒よろしくお願い致します。

また、仰る通り、FP16では結果がおかしくなってしまうことも確認させて頂きました。