2024年9月8日日曜日

rustで顔認識モデルを考える

rust製のMLフレームワークとしてhaggingfaceのcandleを使うとする。



(geminiの回答 2024/09/06)

candleは確かに、PyTorchやTensorFlowと同様に、様々なモデルを扱うことができます。しかし、PyTorchやTensorFlowに比べて、candleで利用できるモデルの数はまだ限られています。

で利用可能なモデル:

  • 基本的なモデル:

    • 線形回帰

    • ロジスティック回帰

    • 畳み込みニューラルネットワーク (CNN)

    • 長短期記憶ネットワーク (LSTM)

    • 循環ニューラルネットワーク (RNN)

  • 事前学習済みモデル:

    • 一部の事前学習済みモデル (例: ImageNetで学習済みのモデル) をロードして使用できます。

  • カスタムモデル:

    • candleのAPIを使って、独自のモデルを定義して学習できます。


<顔検知モデル>

・BlazeFaceを使う

・rustfaceを使う


<顔認識モデル>


candleでPyTorchのモデルを利用する方法:

  1. ONNXへの変換: PyTorchモデルをONNX形式に変換します。

  2. ONNXモデルのロード: candleでONNX形式のモデルをロードして実行します



https://bevyengine.org/

0 件のコメント:

コメントを投稿