classification-tension-for-image 0.0.2

ユーザの顔写真からユーザのテンションを7段階に分類するモデル

このモデルは、ユーザーの顔画像からユーザのテンションを7段階に分類する。このモデルは、ResNet152をベースにしたモデルで、学習済みの重みを用いている。このモデルは、PyTorchを用いて実装されている。 入力：画像のパス 出力：ユーザのテンションを7段階に分類した結果の上位2つのクラスとその確率(これについて変更したい場合は、img_classificationのnum_candidatesを変更してください)

モデルの詳細については以下のリンクを参照してください。(日本語です。) This model classifies user tension into seven levels based on their facial images. It is based on the ResNet152 model and utilizes pretrained weights. The model is implemented using PyTorch. Input: Path to the image Output: Top two classes of user tension levels and their probabilities (to modify this, please change the num_candidates in img_classification)

For more details about the model, please refer to the following link (content is in Japanese).