Skip to content

shreyakate97/fashion-items-classification-and-compatibility

Repository files navigation

fashion-items-classification-and-compatibility

Two tasks have been done: Classification and Compatibilty check

The data is transformed for classification using data.py.

For compatibility check, I have made some changes to the data as explained below.

Data transformation steps for compatibility: data_compat.py

  1. I wrote a python script to create 2 .txt files (train_compatibility.txt and valid_compatibility.txt) where I stored pairs of image ids in each line and the corresponding labels in this order: label image1_id image2_id

  2. I added a function (create_compatibility) in the class polyvore_dataset. It returns X_compatTrain, X_compatValid, Y_compatTrain, Y_compatValid. The X_* contain tuples of the 2 input images (image1.jpg,image2.jpg) and Y_* contain labels.

  3. I changed the getitem functions in classes polyvore_train and polyvore_test so that they returned 2 images (with original transforms applied) and their label.

  4. I made another function get_dataloader_compat to incorporate the above changes to get data.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages