Fine-tuning model for scene classification.

johnolafenwa / DeepStack

The World's Leading Cross Platform AI Engine for Edge Devices

Apache License 2.0

671 stars 106 forks source link

Fine-tuning model for scene classification. #126

Open joaocps opened 2 years ago

joaocps commented 2 years ago

Hello @johnolafenwa !

First of all, I would like to thank you for the excellent work you have done in developing and maintaining this project.

I have a question. Regarding the SceneRecognition API, in docker, is it possible to fine-tune the pre-existing model with my own dataset? For example, run a series of my images, let the model classify them and then use images with a confidence level greater than 85% to fine-tune the model?

Kind Regards, João

gillonba commented 2 years ago

As far as I can tell, there is currently no way to provide a custom model for scene classification. Unfortunately it is anyone's guess whether this will be an option in the future as there do not appear to be regular updates

OlafenwaMoses commented 2 years ago

@gillonba @joaocps Thanks for creating this issue. We do have plans to introduce custom classification models, latest by middle of 2022.