Added custom feature extractor model and a preprocessing function for that model #20
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
A very good and useful repository.
I wanted to use my custom model I trained on a specific dataset to use for feautre extraction for the Reverse Image Search App I am working on.
Deep Image Search library provided everything I needed except the possibility of using my own model. So, I added that functionality to your Library. All the old functionality of the library is there but now you can use your own pretrained model in any format like onnx, torchscript, even tensorflow, or keras or scikit-learn models.
I keep did this by making small changes in selecting the model used for feature extraction. I did not do anything that will increase the dependence on other repos. I will write an example code if you are okay with me to work on this repository with you.
In future I am thinking of adding dimenionality reduction feature for this along with other few changes.
On my side I made sure that the code is backwards compatible. If you find any issues, please let me know.
If we continue to work on this repository together, we can added automated testing to this to make sure it is backwards compatible everything.
If you have any questions or any suggestions, I am open to hearing them.
I also want to use this package in my server, so, I would appreciate if you can publish the changes on PyPi if you deem them to be correct.