值错误:加载预训练模型facenet_keras.h5时封送数据错误

2024-09-29 20:22:50 发布

您现在位置:Python中文网/ 问答频道 /正文

这是我在加载预训练模型facenet_keras.h5时准备数据的代码 错误发生在我加载模型的第16行

import os
import pickle
import numpy as np
import cv2
import mtcnn

from keras.models import load_model
from utils import get_face, get_encode, l2_normalizer, normalize
encoder_model = r'C:\Users\hussa\PycharmProjects\FCN\_03_facenet_keras\data\model\facenet_keras.h5'
people_dir = 'C:/Users/hussa/PycharmProjects/FCN/_03_facenet_keras/data/people'
encodings_path = 'C:/Users/hussa/PycharmProjects/FCN/_03_facenet_keras/data/encodings/encodings.pkl'

required_size = (160, 160)
face_detector = mtcnn.MTCNN()
face_encoder = load_model(encoder_model)
encoding_dict = dict()

for person_name in os.listdir(people_dir):
person_dir = os.path.join(people_dir, person_name)
encodes = []
for img_name in os.listdir(person_dir):
    img_path = os.path.join(person_dir, img_name)
    img = cv2.imread(img_path)
    img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    results = face_detector.detect_faces(img_rgb)
    if results:
        res = max(results, key=lambda b: b['box'][2] * b['box'][3])
        face, _, _ = get_face(img_rgb, res['box'])
        face = normalize(face)
        face = cv2.resize(face, required_size)
        encode = face_encoder.predict(np.expand_dims(face, axis=0))[0]
        encodes.append(encode)
if encodes:
    encode = np.sum(encodes, axis=0)
    encode = l2_normalizer.transform(np.expand_dims(encode, axis=0))[0]
    encoding_dict[person_name] = encode
for key in encoding_dict.keys():
    print(key)
with open(encodings_path, 'bw') as file:
    pickle.dump(encoding_dict, file)

错误消息显示:

"Traceback (most recent call last):

  File "C:/Users/hussa/PycharmProjects/FCN/_03_facenet_keras/01_prepare_data.py", line 16, in <module>
    face_encoder = load_model(encoder_model)

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\saving\save.py", line 182, in load_model
    return hdf5_format.load_model_from_hdf5(filepath, custom_objects, compile)
 
 File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\saving\hdf5_format.py", line 177, in load_model_from_hdf5
    model = model_config_lib.model_from_config(model_config,

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\saving\model_config.py", line 55, in model_from_config
    return deserialize(config, custom_objects=custom_objects)

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\layers\serialization.py", line 171, in deserialize
    return generic_utils.deserialize_keras_object(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\utils\generic_utils.py", line 354, in deserialize_keras_object
    return cls.from_config(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\engine\training.py", line 2238, in from_config
    return functional.Functional.from_config(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\engine\functional.py", line 616, in from_config
    input_tensors, output_tensors, created_layers = reconstruct_from_config(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\engine\functional.py", line 1204, in reconstruct_from_config
    process_layer(layer_data)

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\engine\functional.py", line 1186, in process_layer
    layer = deserialize_layer(layer_data, custom_objects=custom_objects)

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\layers\serialization.py", line 171, in deserialize
    return generic_utils.deserialize_keras_object(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\utils\generic_utils.py", line 354, in deserialize_keras_object
    return cls.from_config(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\layers\core.py", line 1005, in from_config
    function = cls._parse_function_from_config(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\layers\core.py", line 1057, in _parse_function_from_config
    function = generic_utils.func_load(

  File "C:\Users\hussa\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\keras\utils\generic_utils.py", line 457, in func_load
    code = marshal.loads(raw_code)

**ValueError: bad marshal data (unknown type code)**

Process finished with exit code 1"

Tags: infrompyconfigmodelpackageslinesite

热门问题