Added compatibility to new Keras 2.2.0+

[Helias]

Updated the way to import _obtain_input_shape for Keras 2.2.0+

Reference: https://stackoverflow.com/questions/49113140/importerror-cannot-import-name-obtain-input-shape-from-keras