Support for others Other pre-trained models

[33lqq]

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]

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

[virajmavani]

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