Блог

Главная - Вопросы по программированию - использование mask rcnn с keras 2.4.3 в Google colab

использование mask rcnn с keras 2.4.3 в Google colab

#tensorflow #keras #computer-vision #google-colaboratory #image-segmentation

#tensorflow #keras #компьютерное зрение #google-совместная лаборатория #сегментация изображения

Вопрос:

Я пытаюсь запустить Mask-rcnn для обнаружения дефектов металла (https://github.com/maxkferg/metal-defect-detection ) в colab. Версия Keras colab равна 2.4.3, тогда как версия keras репозитория равна 2.1.3. Я получаю следующую ошибку при загрузке весов:

 model.load_weights(model_path, by_name=True, exclude=EXCLUDE_LAYER_WEIGHTS):
AttributeError                            Traceback (most recent call last)
<ipython-input-2-a24cb8cfa577> in <module>()
----> 1 model.load_weights(model_path, by_name=True, exclude=EXCLUDE_LAYER_WEIGHTS)

/content/drive/My Drive/Colab Notebooks/metal-defect-detection/model.py in load_weights(self, filepath, by_name, exclude)
   2030 
   2031         if by_name:
-> 2032             topology.load_weights_from_hdf5_group_by_name(f, layers)
   2033 
   2034         else:

AttributeError: module 'keras.engine.topology' has no attribute 'load_weights_from_hdf5_group_by_name'

Я попробовал некоторые решения, предложенные онлайн, такие как: заменить топологию на сохранение, но у меня та же проблема. Есть способ изменить код, чтобы использовать такой репозиторий с keras 2.4.3?