abhishek/autotrain_fashion_mnist_vit_base
abhishek
Clasificación de imagen
Modelo de Clasificación de Imágenes utilizando Transformadores y PyTorch. Entrenado con AutoTrain en el conjunto de datos fashion_mnist.
Como usar
Cómo usar el modelo
from transformers import ViTForImageClassification, ViTFeatureExtractor
from PIL import Image
import requests
model = ViTForImageClassification.from_pretrained('abhishek/autotrain_fashion_mnist_vit_base')
feature_extractor = ViTFeatureExtractor.from_pretrained('abhishek/autotrain_fashion_mnist_vit_base')
url = 'https://example.com/path/to/image.jpg'
image = Image.open(requests.get(url, stream=True).raw)
inputs = feature_extractor(images=image, return_tensors='pt')
outputs = model(**inputs)
logits = outputs.logits
Funcionalidades
- Clasificación Multi-clase
- Transformadores
- PyTorch
- Entrenado con AutoTrain
Casos de uso
- Clasificación de prendas de moda en imágenes
- Implementación en aplicaciones de comercio electrónico para categorizar productos
- Análisis automatizado de moda