Training Data?

Color Extraction

172
349
MIT
Image-to-Text
PyTorch
English

Opened about a year ago

by @cat-1159

1
@cat-1159a year ago

What dataset did you use to train the model? Also, is it possible to fine-tune it on a custom dataset?

@aiozai_supporta year ago

Hi @cat-1159,
This AI Model in the AI Platform doesn’t use a pre‑trained or trained model with a dataset.
It applies K-Means clustering directly, an unsupervised machine learning technique.

Since it’s not a trained model, there’s no dataset to fine‑tune.
However, we can modify parameters, such as n_clusters, to adapt the color extraction behavior differently.

comment reaction2

0/10000

Sign up or Log in to comment