情绪识别的两个模型

情绪识别的两个模型

tensorflow

https://huggingface.co/CynthiaCR/emotions_classifier

转换为pytorch模型

from transformers import AutoModelForImageClassification
# "./model" clone from CynthiaCR/emotions_classifier on huggingface.
model = AutoModelForImageClassification.from_pretrained("./model", from_tf=True)
model.save_pretrained("./pt_model")

推理

方式1：

import torch
from transformers import AutoModelForImageClassification
from PIL import Image
from transformers import AutoImageProcessor

image = Image.open("./sad01.jpg")

image_processor = AutoImageProcessor.from_pretrained("./pt_model")
inputs = image_processor(image, return_tensors="pt")
# Akhil123/emotions_classifier
model = AutoModelForImageClassification.from_pretrained("./pt_model")
model.eval()
logits = model(**inputs).logits
predicted_class_id = int(torch.argmax(logits, axis=-1)[0])
a = model.config.id2label[predicted_class_id]
print(a)

方式2：

classifier = pipeline("image-classification", model="./model")
x = classifier(image)
s = json.dumps(x, ensure_ascii=False, indent=2)
print(s)

pytorch

https://huggingface.co/Akhil123/emotions_classifier