|
|
@ -401,7 +401,7 @@ class TestBook(unittest.TestCase):
|
|
|
|
self.assertIsNotNone(output)
|
|
|
|
self.assertIsNotNone(output)
|
|
|
|
print(str(program))
|
|
|
|
print(str(program))
|
|
|
|
|
|
|
|
|
|
|
|
def test_maxout(self):
|
|
|
|
def test_crop(self):
|
|
|
|
program = Program()
|
|
|
|
program = Program()
|
|
|
|
with program_guard(program):
|
|
|
|
with program_guard(program):
|
|
|
|
x = layers.data(name='x', shape=[3, 5], dtype="float32")
|
|
|
|
x = layers.data(name='x', shape=[3, 5], dtype="float32")
|
|
|
@ -410,6 +410,15 @@ class TestBook(unittest.TestCase):
|
|
|
|
self.assertIsNotNone(output)
|
|
|
|
self.assertIsNotNone(output)
|
|
|
|
print(str(program))
|
|
|
|
print(str(program))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_mean_iou(self):
|
|
|
|
|
|
|
|
program = Program()
|
|
|
|
|
|
|
|
with program_guard(program):
|
|
|
|
|
|
|
|
x = layers.data(name='x', shape=[16], dtype='float32')
|
|
|
|
|
|
|
|
y = layers.data(name='label', shape=[1], dtype='int64')
|
|
|
|
|
|
|
|
iou = layers.mean_iou(x, y, 2)
|
|
|
|
|
|
|
|
self.assertIsNotNone(iou)
|
|
|
|
|
|
|
|
print(str(program))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|
|
|
|
unittest.main()
|
|
|
|