def test_detection_map(self): program = Program() with program_guard(program): detect_res = layers.data(name='detect_res', shape=[10, 6], append_batch_size=False, dtype='float32') label = layers.data(name='label', shape=[10, 6], append_batch_size=False, dtype='float32') map_out = layers.detection_map(detect_res, label, 21) self.assertIsNotNone(map_out) self.assertEqual(map_out.shape, (1, )) print(str(program))
def test_detection_map(self): program = Program() with program_guard(program): detect_res = layers.data( name='detect_res', shape=[10, 6], append_batch_size=False, dtype='float32') label = layers.data( name='label', shape=[10, 6], append_batch_size=False, dtype='float32') map_out = layers.detection_map(detect_res, label, 21) self.assertIsNotNone(map_out) self.assertEqual(map_out.shape, (1, )) print(str(program))