Как я могу загрузить файл весов с n классов в модели CNN с n + 1 классами

Question

Я построил модель CNN с 3 метками классов, а также сгенерировал для нее вес (файл .h5). Я хочу использовать этот файл весов с той же моделью, но единственное отличие состоит в том, что он теперь имеет 4 класса вместо 3.

Я пытался использовать

checkpoint = model.load_weights("filename.h5") 

но я всегда получаю ошибку как несоответствие класса.

Может ли кто-нибудь помочь мне с любым предложением

Answer 1

Может быть, следуйте совету yanboliang в Keras: удаление слоев с помощью model.layers.pop () не работает , перезаписывая последний слой модели новым слоем с выходами из 4 классов?