как загрузить обученную модель .pb vgg16

Question

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

# load network
network_pkl = misc.locate_network_pkl(args.results_dir)
print('Loading network from "%s"...' % network_pkl)
G, D, Gs = misc.load_network_pkl(args.results_dir, None)



def load_network_pkl(run_id_or_result_subdir_or_network_pkl, snapshot=None):
    return load_pkl(locate_network_pkl(run_id_or_result_subdir_or_network_pkl, snapshot))


def load_pkl(filename):
    with open(filename, 'rb') as file:
         return legacy.LegacyUnpickler(file, encoding='latin').load()

def save_pkl(obj, filename):
    with open(filename, 'wb') as file:
        pickle.dump(obj, file, protocol=pickle.HIGHEST_PROTOCOL)

, и я получил эту ошибку, как показано ниже:

Инициализация TensorFlow ... Загрузка сети из "C: \ Users \ wissa \ Desktop \ Out-source-projects / vgg_tf.pb" ... Traceback (последний вызов был последним): файл "beautify_image.py", строка 46, в G, D, Gs = mis c .load_network_pkl (args.results_dir, None) Файл "C: \ Users \ wissa \ Desktop \ Out-source-projects \ Beholder-GAN-master \ mis c. py ", строка 674, в load_network_pkl возвращает load_pkl (locate_network_pkl (run_id_or_result_subdir_or_network_pkl, снимок)) Файл" C: \ Users \ wissa \ Desktop \ Out-source-projects \ Beholder-GAN-master \ mis cpy. , строка 436, в load_pkl вернуть pickle.load (файл) _pickle.UnpicklingError: неверный ключ загрузки, ',' '.

Не могли бы вы помочь мне исправить это?