Imported from https://github.com/jsbruglie/tf-shape-detection