[ResearchAtPlay]

This is great! I've forwarded the site to my wife.

Would you mind sharing how you trained the model to produce the vectors? Are you using a vision transformer under the hood with contrastive training against price, product category, etc.?

EDIT: I see that the training script is included in the repo and you are using a CNN. Inspiring work!

[unixpickle]

Yup, it's a small model I trained on my Mac mini! The model itself just classifies product attributes like keywords, price, retailer, etc. The features it learns are then used as embeddings