anismizi/skin-type-classifier

🔬 Skin Type Classification Model

A deep learning model for classifying skin types into dry and oily categories using computer vision.

Model Description

This model is based on ResNet50 architecture and has been fine-tuned specifically for skin type classification. It can analyze facial skin images and determine whether the skin type is dry or oily with high accuracy.

Key Features

• Architecture: ResNet50-based classification model
• Classes: 2 (dry, oily)
• Input: RGB images (224x224 pixels)
• Framework: PyTorch + Transformers
• Performance: High accuracy on skin type classification

Intended Use

Primary Use Cases

• Dermatological analysis and skin assessment
• Cosmetic product recommendation systems
• Skincare routine personalization
• Medical research and skin health monitoring

Limitations

• Designed specifically for facial skin analysis
• Requires good lighting and clear skin visibility
• Not suitable for medical diagnosis (for research/cosmetic use only)
• Performance may vary across different skin tones and ethnicities

How to Use

Quick Start with Transformers

from transformers import AutoModelForImageClassification, AutoImageProcessor
from PIL import Image
import torch

# Load model and processor
model = AutoModelForImageClassification.from_pretrained("your-username/skin-type-classifier")
processor = AutoImageProcessor.from_pretrained("your-username/skin-type-classifier")

# Load and process image
image = Image.open("path/to/skin/image.jpg")
inputs = processor(images=image, return_tensors="pt")

# Make prediction
with torch.no_grad():
    outputs = model(**inputs)
    predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
    predicted_class = predictions.argmax().item()

# Get result
labels = ["dry", "oily"]
confidence = predictions[0][predicted_class].item()
print(f"Predicted skin type: {labels[predicted_class]} (confidence: {confidence:.2%})")

Using the Pipeline API

from transformers import pipeline

# Create classification pipeline
classifier = pipeline("image-classification", model="your-username/skin-type-classifier")

# Classify image
result = classifier("path/to/skin/image.jpg")
print(result)

Model Details

Architecture

• Base Model: ResNet50
• Modification: Custom classification head with 2 output classes
• Input Size: 224 × 224 × 3 (RGB)
• Parameters: ~25M parameters

Training Details

• Dataset: Custom skin type classification dataset
• Preprocessing:
  • Resize to 224×224 pixels
  • Normalization: ImageNet statistics
  • Data augmentation applied during training
• Training Framework: PyTorch
• Optimization: Adam optimizer with learning rate scheduling

Performance Metrics

• Training Accuracy: High performance on validation set
• Inference Speed: Fast inference suitable for real-time applications
• Model Size: ~94MB

Technical Specifications

Input Format

• Type: RGB Images
• Size: 224 × 224 pixels
• Format: PIL Image, numpy array, or torch tensor
• Normalization: ImageNet mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]

Output Format

• Type: Classification logits
• Classes:
  • 0: "dry" - Dry skin type
  • 1: "oily" - Oily skin type
• Output: Softmax probabilities for each class

Ethical Considerations

Bias and Fairness

• Model trained on diverse skin types but may have limitations
• Users should be aware of potential biases in skin tone representation
• Continuous evaluation needed for fair performance across demographics

Privacy

• Model processes images locally - no data transmission required
• Users responsible for ensuring proper consent when analyzing others' images
• Recommend anonymization of facial features when possible

License

This model is released under the MIT License. See LICENSE file for details.

Citation

If you use this model in your research, please cite:

@misc{skin-type-classifier-2025,
  title={Skin Type Classification Model},
  author={Your Name},
  year={2025},
  howpublished={\\url{https://huggingface.co/your-username/skin-type-classifier}},
}

Contact

For questions, issues, or collaboration opportunities, please reach out through the Hugging Face model page or GitHub repository.

---

Disclaimer: This model is for research and cosmetic purposes only. It should not be used for medical diagnosis or treatment decisions. Always consult healthcare professionals for medical concerns.