You can deploy this model and start earning money today!

Iris314/
nn_automl_model

Name: nn_automl_model
Availability: InStock

keerthikoganti/lipstick-image-dataset

model-index

Views

No views yet

Model Card for Lipstick Detection (Neural Network AutoML)

This model performs binary classification of images into lipstick (1) vs. no lipstick (0).
It was trained with AutoGluon Multimodal AutoML, which automatically explored different neural network backbones (ResNet18, ResNet34, EfficientNet-B0) under a fixed budget with early stopping.
The best-performing backbone selected was EfficientNet-B0.

Model Details

Model Description

Developed by: Xinxuan Tang (CMU)
Dataset curated by: Keerthi Koganti (CMU)
Model type: AutoML neural network (best = EfficientNet-B0)
Language(s): N/A (image dataset)
Finetuned from: timm/efficientnet_b0 pretrained weights

Model Sources

Repository: Hugging Face Model Repo
Dataset: keerthikoganti/lipstick-image-dataset

Uses

Direct Use

Educational practice in binary image classification.
Experimenting with AutoML search over neural architectures.

Downstream Use

Could be adapted for teaching transfer learning workflows.

Out-of-Scope Use

Not suitable for real-world cosmetics applications.
Not for deployment in automated decision-making or safety-critical contexts.

Bias, Risks, and Limitations

Small dataset: limited original images, heavy reliance on synthetic augmentation.
Domain bias: images are from a single source/product and background setup.
Synthetic augmentation: does not capture real-world variation in lighting, product types, or diversity of appearances.

Recommendations

Use primarily for teaching and demonstration purposes.
Do not generalize conclusions beyond this dataset.

How to Get Started with the Model

python

from autogluon.multimodal import MultiModalPredictor
import pandas as pd

# Load trained predictor
predictor = MultiModalPredictor.load("autogluon_efficientnet_b0/")

# Run inference on a new image
test_data = pd.DataFrame([{"image": "example.jpg"}])
print(predictor.predict(test_data))

Iris314/nn_automl_model