Python create_3d_image_test_example Beispiele

Beispiel #1

Datei: segmentation_input_3d_test.py Projekt: shizukanaskytree/models

 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)

Beispiel #2

Datei: semantic_segmentation_3d_test.py Projekt: xiangww00/models

 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)