Caplin43 commited on
Commit
3bc795a
·
verified ·
1 Parent(s): 7ae0aff

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - robotics
5
+ - humanoid
6
+ - action-recognition
7
+ - deep-learning
8
+ - computer-vision
9
+ pipeline_tag: image-classification
10
+ ---
11
+
12
+ # Humanoid Action Classifier v1
13
+
14
+ This model is designed for humanoid robot action recognition.
15
+ It classifies basic humanoid movements from image input.
16
+
17
+ ## Supported Actions
18
+ - walk
19
+ - run
20
+ - sit
21
+ - stand
22
+ - wave
23
+ - pick_object
24
+ - turn_left
25
+ - turn_right
26
+
27
+ ## Model Details
28
+ - Architecture: CNN (ResNet18 backbone)
29
+ - Framework: PyTorch
30
+ - Input Size: 224x224 RGB
31
+ - Output: 8 action classes
32
+
33
+ ## Training Info
34
+ Trained on a synthetic humanoid action dataset.
35
+ Optimized for robotics simulation environments.
36
+
37
+ ## Usage
38
+
39
+ ```python
40
+ from transformers import AutoModelForImageClassification
41
+ from PIL import Image
42
+ import torch
43
+
44
+ model = AutoModelForImageClassification.from_pretrained("your-username/humanoid-action-classifier-v1")
45
+
46
+ image = Image.open("test.jpg")
47
+ inputs = processor(images=image, return_tensors="pt")
48
+
49
+ with torch.no_grad():
50
+ outputs = model(**inputs)
51
+
52
+ predicted_class = outputs.logits.argmax(-1)