| # Copyright 2018 The TensorFlow Authors. All Rights Reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # ============================================================================== |
| """Utilities for serializing Python objects.""" |
| |
| from __future__ import absolute_import |
| from __future__ import division |
| from __future__ import print_function |
| |
| import collections |
| |
| import numpy as np |
| |
| from tensorflow.python.framework import tensor_shape |
| |
| |
| def get_json_type(obj): |
| """Serializes any object to a JSON-serializable structure. |
| |
| Arguments: |
| obj: the object to serialize |
| |
| Returns: |
| JSON-serializable structure representing `obj`. |
| |
| Raises: |
| TypeError: if `obj` cannot be serialized. |
| """ |
| # if obj is a serializable Keras class instance |
| # e.g. optimizer, layer |
| if hasattr(obj, 'get_config'): |
| return {'class_name': obj.__class__.__name__, 'config': obj.get_config()} |
| |
| # if obj is any numpy type |
| if type(obj).__module__ == np.__name__: |
| if isinstance(obj, np.ndarray): |
| return obj.tolist() |
| else: |
| return obj.item() |
| |
| # misc functions (e.g. loss function) |
| if callable(obj): |
| return obj.__name__ |
| |
| # if obj is a python 'type' |
| if type(obj).__name__ == type.__name__: |
| return obj.__name__ |
| |
| if isinstance(obj, tensor_shape.Dimension): |
| return obj.value |
| |
| if isinstance(obj, tensor_shape.TensorShape): |
| return obj.as_list() |
| |
| if isinstance(obj, collections.Mapping): |
| return dict(obj) |
| |
| raise TypeError('Not JSON Serializable:', obj) |