test_mask_to_seq.py 文件源码

def test_image_data_mask(self):
        mask_cache_key = str(id(self.model.input)) + '_' + str(id(None))
        mask_tensor = self.model._output_mask_cache[mask_cache_key]
        mask = mask_tensor.eval(
            session=K.get_session(),
            feed_dict={self.model.input: self.data}
        )
        self.assertTrue(
            np.all(
                mask[:, self.x_start:self.x_end]
            )
        )
        self.assertFalse(
            np.any(
                mask[:, :self.x_start]
            )
        )
        self.assertFalse(
            np.any(
                mask[:, self.x_end:]
            )
        )