search

akarshzingade / image-similarity-deep-ranking

369 stars 103 forks source link

Generate a pkl file to save filenames and classes for reuse #20

Closed cesarandreslopez closed 6 years ago

cesarandreslopez commented 6 years ago

When ImageDataGeneratorCustom is called and a lot of images and classes are used in training (i.e. market 1501 dataset) loading of classes and filenames can be quite long.

Hear, I save those on a pickle file for use next time deepRanking.py is called, in particular when the triplets generated have not changed.

If triplets are new, then generated 'delete_if_new_data_source.pkl' should be deleted.

cesarandreslopez commented 6 years ago

Will submit another pull request including this and other modifications, please ignore.

Apologies.