experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test case for 28241.

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2019-04-30 01:09:51 +00:00
parent 6c3cbe4434
commit 457f1c10ca
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/ops/image_ops_test.py
View File

@ -4860,6 +4860,12 @@ class MultiscaleSSIMTest(test_util.TensorFlowTestCase):
self.assertAllClose( self.assertAllClose(
ssim_uint8.eval(), self.evaluate(ssim_float32), atol=0.001) ssim_uint8.eval(), self.evaluate(ssim_float32), atol=0.001)
def testNumpyInput(self):
"""Test case for GitHub issue 28241."""
image = np.random.random([512, 512, 1])
score_tensor = image_ops.ssim_multiscale(image, image, max_val=1.0)
with self.cached_session(use_gpu=True):
_ = self.evaluate(score_tensor)
class ImageGradientsTest(test_util.TensorFlowTestCase): class ImageGradientsTest(test_util.TensorFlowTestCase):