search

virajmavani / semi-auto-image-annotation-tool

Anno-Mage: A Semi Automatic Image Annotation Tool which helps you in annotating images by suggesting you annotations for 80 object classes using a pre-trained model
https://www.virajmavani.com/saiat
Apache License 2.0
578 stars 127 forks source link

Support for others Other pre-trained models #11

Open 33lqq opened 5 years ago

33lqq commented 5 years ago

Hi，I am doing garbage identification, not in your 80 categories, like stone，banana peel ...So I want to use my pre-trained model, which is a pb file, so what am I trying to replace? I sincerely hope that you can answer my question.

33lqq commented 5 years ago

Can I directly change the weight h5 file to my pb file? Then modify the config.py file

virajmavani commented 4 years ago

@33lqq Well this doesn't currently support other pretrained models. It is definitely a nice feature that can be added.