Próbuję pobrać ten model, zawierający niestandardowe funkcje i lambda-warstwy za pomocą następującego kodu:
def load_efficient_pose_model():
custom_objects = {'BilinearWeights': keras_BilinearWeights, 'Swish': Swish(eswish), 'eswish': eswish, 'swish1': swish1}
model = keras.models.load_model('my_location/EfficientPoseII.h5',custom_objects=custom_objects)
Podczas korzystania z Tensorflow 1.13.1 i Keras wszystko działa poprawnie. Jednak, gdy używam Tensorflow 2.7 (z wbudowanym Keras), występuje następujący błąd:
Traceback (most recent call last):
....
File "/home/ubuntu/anaconda3/envs/tf2_runtime/lib/python3.7/site-packages/keras/utils/traceback_utils.py", line 67, in error_handler
raise e.with_traceback(filtered_tb) from None
File "/Users/daniegr/Programming/EfficientPose/utils/units.py", line 90, in <lambda>
NameError: name 'K' is not defined
Próbowałem również pobrać model w Tensorflow 1.13.1 i zapisz model i masy ciała oddzielnie, ale problem pozostaje.
Podczas serializacji modelu w formacie JSON dostaję te ciekawe wiersze:
{ "name": "lambda_1_res1", "class_name": "Lambda", "config": { "name": "lambda_1_res1", "trainable": false, "function": [ "4wEAAAAAAAAAAQAAAAEAAABTAAAAcwQAAAB8AFMAKQFOqQApAdoBeHIBAAAAcgEAAAB6Ri9Vc2Vy\ncy9kYW5pZWdyL1Byb2dyYW1taW5nL0VmZmljaWVudFBvc2UvbW9kZWxzL2VmZmljaWVudG5ldHMv\nbW9kZWwucHnaCDxsYW1iZGE+HQEAAPMAAAAA\n", null, null ], "function_type": "lambda", "output_shape": null, "output_shape_type": "raw", "arguments": {} },
Zakładam, że w jakiś sposób stary model zachowuje części kodu, które uzyskują dostęp K jako zaplecza części, która nie jest dostępna w Tensorflow 2.
Przejrzałem te związane z tym pytania so1 i so2, ale nie mogłem znaleźć odpowiedzi.
Jakieś pomysły?