Fashion attribute and category prediction.
-
Updated
Oct 22, 2020 - Python
Fashion attribute and category prediction.
A multi-label, multi-class classification project on the DeepFashion Dataset. We fine-tuned a SE-ResNeXt model and optimized hyper-parameters with Optuna. By employing a diffusion model to generate images for underrepresented classes, we improved dataset balance and model performance.
Add a description, image, and links to the deep-fashion topic page so that developers can more easily learn about it.
To associate your repository with the deep-fashion topic, visit your repo's landing page and select "manage topics."