Add batch support for various image_ops (#14854)
* Change fix_image_flip_shape to create shape based on rank * Refactor duplicate code to _EnsureTensorIs4D * Convert flip_up_down * Temporarily comment out ValueError Check * Add batch support for flip_left_right * Add batch support for random_flip_left_right * Add batch support for random_flip_up_down * Add batch support for transpose_image * Add batch support for rot90 * Correct comments * Refactor so as not to introduce new method * Add tests for batch inputs * Fix test to expect 3 or 4 dims * Fix misc Pylint issues in image_ops_impl.py * Fix misc Pyline issues in image_ops_test.py * Refactor into _flip_image * Correct Idempotent to Involution * Check if >20 images were flipped * Reverse condition in rot90 * Remove duplicate comment * Address feedback * Punctuation
Loading
Please sign in to comment