Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Can I recognise the old and new classes on one re-trained image classification model? #1880

Open
Vuforiator opened this issue Jul 24, 2024 · 0 comments

Comments

@Vuforiator
Copy link

I’m completely new to the topic “Hello AI World/ Deploying Deep Learning” and perhaps my question will seem stupid, but I’ve been trying to find the answer for days, but none of those around me can answer. The problem I have is : I want to recognise my own ("new") images using pre-trained image classification model.

  • I created my own datasets for several new classes, I classify the Raspberry Pi series microcomputers (B3, B3+, B4 etc) - OK;
  • I used the train.py utility to retrain the one of the pre-trained image classification models and convert the result into ONNX-format – OK;
  • I used imagenet.py with onnx-file for recognition my microcomputers and I can recognize every of RaspberryPis under cаmera with some confidence – OK
    But I cannot recognize the “old” classes for the pre-trained image classification model/ I used both ResNet18 and GooleNet – the problem is the same: after retraining, none of the converted models could recognize any other images except images of my objects, the network only worked with my new classes and did not work with its “old” ones (apple, watermelon, computer, etc.)
    My question: does it possible to use imagenet.py for recognize both “old” and “new” (my own) classes on one re-trained image classification model?
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant