yolo-human-parse / README.yaml
MnLgt's picture
hi
bb51988
raw
history blame
No virus
2.77 kB
---
language:
- "en"
thumbnail: "https://example.com/path/to/your/thumbnail.jpg"
tags:
- yolo
- object-detection
- image-segmentation
- computer-vision
- human-body-parts
license: "mit"
datasets:
- custom_human_body_parts_dataset
metrics:
- mean_average_precision
- intersection_over_union
base_model: "ultralytics/yolov5yolov8x-seg"
---
# YOLO Segmentation Model for Human Body Parts and Objects
This model is a fine-tuned version of YOLOv5 for segmenting human body parts and objects. It can detect and segment 11 different classes including various body parts, outfits, and phones.
## Model Details
- **Model Type:** YOLOv8 for Instance Segmentation
- **Task:** Segmentation
- **Fine-tuning Dataset:** Custom dataset of human body parts and objects
- **Number of Classes:** 11
## Classes
The model can detect and segment the following classes:
0. Hair
1. Face
2. Neck
3. Arm
4. Hand
5. Back
6. Leg
7. Foot
8. Outfit
9. Person
10. Phone
## Usage
This model can be used for various applications, including:
- Human pose estimation
- Gesture recognition
- Fashion analysis
- Person tracking
- Human-computer interaction
For detailed usage instructions, please refer to the model's README file.
## Training Procedure
The model was fine-tuned on a custom dataset of annotated images containing human body parts and objects. The training process involved transfer learning from the base YOLOv8 model, with adjustments made to the final layers to accommodate the new class structure.
## Evaluation Results
(Note: Replace these placeholder metrics with your actual evaluation results)
lr/pg0:0.000572628
lr/pg1:0.000572628
lr/pg2:0.000572628
metrics/mAP50-95(B):0.53001
metrics/mAP50-95(M):0.42367
metrics/mAP50(B):0.69407
metrics/mAP50(M):0.61714
metrics/precision(B):0.7047
metrics/precision(M):0.68041
metrics/recall(B):0.68802
metrics/recall(M):0.62248
model/GFLOPs:344.557
model/parameters:71,761,441
model/speed_PyTorch(ms):5.813
train/box_loss:0.54718
train/cls_loss:0.52977
train/dfl_loss:0.95171
train/seg_loss:1.34628
val/box_loss:0.80538
val/cls_loss:0.83434
val/dfl_loss:1.18352
val/seg_loss:2.19488
## Limitations and Biases
- The model's performance may vary depending on lighting conditions and image quality.
- It may have difficulty with occluded or partially visible body parts.
- The model's performance on diverse body types and skin tones should be carefully evaluated to ensure fairness and inclusivity.
## Ethical Considerations
Users of this model should be aware of privacy concerns related to human body detection and ensure they have appropriate consent for its application. The model should not be used for surveillance or any application that could infringe on personal privacy without explicit consent.