Сейчас я борюсь за запуск моего первого проекта, SRGAN. Я получил код от Github, который был размещен на Jupyter nbviewer, и здесь есть все необходимое для работы, включая веса. Кажется, что каждый код чист, и при запуске других файлов ошибок не возникает. Но проблема заключается в загрузке весов.
Как написано в предварительном просмотре, сначала я попытался загрузить предварительно обученные веса из zip-файла, который находится на заданных данных. Во-вторых, я распаковал весовой файл и просто использовал model.load_weights(file path). Но обе попытки заканчиваются неудачей с комментарием об ошибке
ImportError: cannot import name 'load_weights' from 'tensorflow.python.keras.models'
Хотя я изменил каждое «обучаемое» слово на ложное, но оно все равно не работает.
В виде файла я закодировал, как показано ниже. и я прикрепляю Git источника вместо того, чтобы писать остальные коды. Ах, я просто использую SRGAN, а не другие модели тонкой настройки.
https://github.com/krasserm/super-resolution/blob/master/article.ipynb
Спасибо.
import h5py
from tensorflow.python.keras.models
import load_weights
from model.srgan import generator
generator = generator()
generator.load_weights("/weights_srgan/weights/srgan/gan_generator.h5")