Python Config.from_json Beispiele

Beispiel #1

Datei: test_saved_model_to_openvino_model.py Projekt: jiangjiajun/Adlik

 def test_compile_with_all_params_with_enable_nhwc_to_nchw_true(self):
     config = Config.from_json({
         'input_names': ['x', 'y'],
         'output_names': ['z'],
         'max_batch_size': 1,
         'enable_nhwc_to_nchw': True
     })
     compiled = compiler.compile_source(source=_make_saved_model(),
                                        config=config)
     self.assertEqual(
         compiled.get_inputs(),
         [
             ModelInput(name='x',
                        data_type=tf.float32.as_datatype_enum,
                        format=ModelInput.FORMAT_NONE,
                        dims=[4, 2, 3]),  # pylint: disable=no-member
             ModelInput(name='y',
                        data_type=tf.float32.as_datatype_enum,
                        format=ModelInput.FORMAT_NONE,
                        dims=[4, 2, 3])
         ])  # pylint: disable=no-member
     self.assertEqual(compiled.get_outputs(), [
         ModelOutput(name='z',
                     data_type=tf.float32.as_datatype_enum,
                     dims=[4, 2, 3])
     ])

Beispiel #2

Datei: test_saved_model_to_openvino_model.py Projekt: ouyufeng/Adlik

 def test_compile_with_no_params(self):
     config = Config.from_json({'max_batch_size': 1})
     compiled = compiler.compile_source(source=_make_saved_model(), config=config)
     self.assertEqual(compiled.get_inputs(),
                      [ModelInput(name='x', data_type=tf.float32.as_datatype_enum,
                                  format=ModelInput.FORMAT_NONE, dims=[2, 3, 4]),  # pylint: disable=no-member
                       ModelInput(name='y', data_type=tf.float32.as_datatype_enum,
                                  format=ModelInput.FORMAT_NONE, dims=[2, 3, 4])])  # pylint: disable=no-member
     self.assertEqual(compiled.get_outputs(),
                      [ModelOutput(name='z', data_type=tf.float32.as_datatype_enum, dims=[2, 3, 4])])