Downloads · 30 days
20
0% of all-time downloads
glazzova/body_type
body_type is a image classification model from glazzova. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Эта модель дообучена на microsoft/resnet-50 с помощью датасета, который содержит фотографии мужчин разных телосложений. Модель может определяет являетесь ли вы качком, скуфом, дрищом или просто нормальным.
Downloads · 30 days
20
0% of all-time downloads
All-time downloads
72.9K
Public
Repo size
377 MB
Likes
7
Public
Click a slice to open those files.
.pt188 MB · 67%
From the Hugging Face model README
Эта модель дообучена на microsoft/resnet-50 с помощью датасета, который содержит фотографии мужчин разных телосложений. Модель может определяет являетесь ли вы качком, скуфом, дрищом или просто нормальным.
import torch
from PIL import Image
from transformers import ResNetForImageClassification, AutoImageProcessor
processor = AutoImageProcessor.from_pretrained("glazzova/body_type")
model = ResNetForImageClassification.from_pretrained('glazzova/body_type')
image = Image.open('your_pic.jpeg')
inputs = processor(image, return_tensors="pt")
with torch.no_grad():
logits = model(**inputs).logits
# model predicts one of the 4 classes
predicted_label = logits.argmax(-1).item()
print(model.config.id2label[predicted_label])