Caplin43/humanoid-action-classifier-v1

Humanoid Action Classifier v1

This model is designed for humanoid robot action recognition. It classifies basic humanoid movements from image input.

Supported Actions

• —walk
• —run
• —sit
• —stand
• —wave
• —pick_object
• —turn_left
• —turn_right

Model Details

• —Architecture: CNN (ResNet18 backbone)
• —Framework: PyTorch
• —Input Size: 224x224 RGB
• —Output: 8 action classes

Training Info

Trained on a synthetic humanoid action dataset. Optimized for robotics simulation environments.

Usage

from transformers import AutoModelForImageClassification
from PIL import Image
import torch

model = AutoModelForImageClassification.from_pretrained("your-username/humanoid-action-classifier-v1")

image = Image.open("test.jpg")
inputs = processor(images=image, return_tensors="pt")

with torch.no_grad():
    outputs = model(**inputs)

predicted_class = outputs.logits.argmax(-1)