def setUp(self): super().setUp() data_dir = os.path.join(self.get_temp_dir(), 'data') tf.io.gfile.makedirs(data_dir) self._data_path = os.path.join(data_dir, 'data.tfrecord') self._example = tfexample_utils.create_3d_image_test_example( image_height=32, image_width=32, image_volume=32, image_channel=2)
def setUp(self): super().setUp() data_dir = os.path.join(self.get_temp_dir(), 'data') tf.io.gfile.makedirs(data_dir) self._data_path = os.path.join(data_dir, 'data.tfrecord') # pylint: disable=g-complex-comprehension examples = [ tfexample_utils.create_3d_image_test_example( image_height=32, image_width=32, image_volume=32, image_channel=2) for _ in range(20) ] # pylint: enable=g-complex-comprehension tfexample_utils.dump_to_tfrecord(self._data_path, tf_examples=examples)