Пример #1
0
 def testGlimpseNonNormalizedNonCentered(self):
     img = constant_op.constant(np.arange(25).reshape((1, 5, 5, 1)),
                                dtype=dtypes.float32)
     with self.test_session():
         result1 = image_ops.extract_glimpse_v2(img, [3, 3], [[0, 0]],
                                                centered=False,
                                                normalized=False)
         result2 = image_ops.extract_glimpse_v2(img, [3, 3], [[1, 0]],
                                                centered=False,
                                                normalized=False)
         self.assertAllEqual(
             np.asarray([[0, 1, 2], [5, 6, 7], [10, 11, 12]]),
             self.evaluate(result1)[0, :, :, 0])
         self.assertAllEqual(
             np.asarray([[5, 6, 7], [10, 11, 12], [15, 16, 17]]),
             self.evaluate(result2)[0, :, :, 0])
Пример #2
0
 def testGlimpseNegativeInput(self):
     img = np.arange(9).reshape([1, 3, 3, 1])
     with self.test_session():
         with self.assertRaises((errors.InvalidArgumentError, ValueError)):
             result = image_ops.extract_glimpse_v2(img,
                                                   size=[1023, -63],
                                                   offsets=[1023, 63],
                                                   centered=False,
                                                   normalized=False)
             self.evaluate(result)
Пример #3
0
    def testGlimpseNoiseZero(self):
        # Image:
        # [  0.   1.   2.   3.   4.]
        # [  5.   6.   7.   8.   9.]
        # [ 10.  11.  12.  13.  14.]
        # [ 15.  16.  17.  18.  19.]
        # [ 20.  21.  22.  23.  24.]
        img = constant_op.constant(np.arange(25).reshape((1, 5, 5, 1)),
                                   dtype=dtypes.float32)
        with self.test_session():
            # Result 1:
            # [ 0.  0.  0.]
            # [ 0.  0.  0.]
            # [ 0.  0.  0.]
            result1 = image_ops.extract_glimpse_v2(img, [3, 3], [[-2, 2]],
                                                   centered=False,
                                                   normalized=False,
                                                   noise='zero')
            self.assertAllEqual(np.asarray([[0, 0, 0], [0, 0, 0], [0, 0, 0]]),
                                self.evaluate(result1)[0, :, :, 0])

            # Result 2:
            # [  0.   0.   0.   0.   0.   0.   0.]
            # [  0.   0.   1.   2.   3.   4.   0.]
            # [  0.   5.   6.   7.   8.   9.   0.]
            # [  0.  10.  11.  12.  13.  14.   0.]
            # [  0.  15.  16.  17.  18.  19.   0.]
            # [  0.  20.  21.  22.  23.  24.   0.]
            # [  0.   0.   0.   0.   0.   0.   0.]]
            result2 = image_ops.extract_glimpse_v2(img, [7, 7], [[0, 0]],
                                                   normalized=False,
                                                   noise='zero')
            self.assertAllEqual(
                np.asarray([[0, 0, 0, 0, 0, 0, 0], [0, 0, 1, 2, 3, 4, 0],
                            [0, 5, 6, 7, 8, 9, 0], [0, 10, 11, 12, 13, 14, 0],
                            [0, 15, 16, 17, 18, 19, 0],
                            [0, 20, 21, 22, 23, 24, 0], [0, 0, 0, 0, 0, 0,
                                                         0]]),
                self.evaluate(result2)[0, :, :, 0])
Пример #4
0
  def testGlimpseNoiseZero(self):
    # Image:
    # [  0.   1.   2.   3.   4.]
    # [  5.   6.   7.   8.   9.]
    # [ 10.  11.  12.  13.  14.]
    # [ 15.  16.  17.  18.  19.]
    # [ 20.  21.  22.  23.  24.]
    img = constant_op.constant(
        np.arange(25).reshape((1, 5, 5, 1)), dtype=dtypes.float32)
    with self.test_session():
      # Result 1:
      # [ 0.  0.  0.]
      # [ 0.  0.  0.]
      # [ 0.  0.  0.]
      result1 = image_ops.extract_glimpse_v2(
          img, [3, 3], [[-2, 2]],
          centered=False,
          normalized=False,
          noise='zero')
      self.assertAllEqual(
          np.asarray([[0, 0, 0], [0, 0, 0], [0, 0, 0]]),
          self.evaluate(result1)[0, :, :, 0])

      # Result 2:
      # [  0.   0.   0.   0.   0.   0.   0.]
      # [  0.   0.   1.   2.   3.   4.   0.]
      # [  0.   5.   6.   7.   8.   9.   0.]
      # [  0.  10.  11.  12.  13.  14.   0.]
      # [  0.  15.  16.  17.  18.  19.   0.]
      # [  0.  20.  21.  22.  23.  24.   0.]
      # [  0.   0.   0.   0.   0.   0.   0.]]
      result2 = image_ops.extract_glimpse_v2(
          img, [7, 7], [[0, 0]], normalized=False, noise='zero')
      self.assertAllEqual(
          np.asarray([[0, 0, 0, 0, 0, 0, 0], [0, 0, 1, 2, 3, 4, 0],
                      [0, 5, 6, 7, 8, 9, 0], [0, 10, 11, 12, 13, 14, 0],
                      [0, 15, 16, 17, 18, 19, 0], [0, 20, 21, 22, 23, 24, 0],
                      [0, 0, 0, 0, 0, 0, 0]]),
          self.evaluate(result2)[0, :, :, 0])