Object_classification and Image similarity..
Task 1
"To build an image recognition model which is capable of identifying the pattern on a dress image.”
You can get the script is in final_model folder.
Dataset Information :
- This dataset contains links to images of women's dresses.
- The corresponding images are categorized into 17 different pattern
types.
- Most pattern categories have hundreds to thousands of examples.
Trained Model download link
Training data
Run the script 'train_vgg19.py'
Testing data
Run the script 'test_vgg19.py'
Task 2
Download data
"Given a random query image we need to return the similar products.”
You can get the script in images_similarity folder.
To create cnn feature for data
- Run the script 'create_cnn_feature.py'
- Save the image cnn feature and it's corresponding label
Testing
- Run the script 'images_similarity.py' by passing random image and cnn image feature and label path