| import os |
| import keras |
| from keras.models import load_model |
| |
| |
| class ModelSerializer(object): |
| |
| def _serializer_load(self, object_file_path): |
| if object_file_path.split(os.sep)[-1] == 'model': |
| keras.backend.clear_session() |
| return load_model(object_file_path) |
| else: |
| return super(ModelSerializer, self)._serializer_load(object_file_path) |
| |
| def _serializer_dump(self, obj, object_file_path): |
| if object_file_path.split(os.sep)[-1] == 'model': |
| obj.save(object_file_path) |
| else: |
| super(ModelSerializer, self)._serializer_dump(obj, object_file_path) |
