Python get_model_variables示例

示例#1

文件： variables_test.py 项目： AliMiraftab/tensorflow

 def testGetModelVariables(self):
   with self.test_session():
     with variable_scope.variable_scope('A'):
       a = variables_lib2.model_variable('a', [5])
     with variable_scope.variable_scope('B'):
       b = variables_lib2.model_variable('a', [5])
     self.assertEquals([a], variables_lib2.get_model_variables('A'))
     self.assertEquals([b], variables_lib2.get_model_variables('B'))

示例#2

文件： overfeat_test.py 项目： AliMiraftab/tensorflow

 def testModelVariables(self):
   batch_size = 5
   height, width = 231, 231
   num_classes = 1000
   with self.test_session():
     inputs = random_ops.random_uniform((batch_size, height, width, 3))
     overfeat.overfeat(inputs, num_classes)
     expected_names = [
         'overfeat/conv1/weights',
         'overfeat/conv1/biases',
         'overfeat/conv2/weights',
         'overfeat/conv2/biases',
         'overfeat/conv3/weights',
         'overfeat/conv3/biases',
         'overfeat/conv4/weights',
         'overfeat/conv4/biases',
         'overfeat/conv5/weights',
         'overfeat/conv5/biases',
         'overfeat/fc6/weights',
         'overfeat/fc6/biases',
         'overfeat/fc7/weights',
         'overfeat/fc7/biases',
         'overfeat/fc8/weights',
         'overfeat/fc8/biases',
     ]
     model_variables = [v.op.name for v in variables_lib.get_model_variables()]
     self.assertSetEqual(set(model_variables), set(expected_names))

示例#3

文件： alexnet_test.py 项目： Ajaycs99/tensorflow

 def testModelVariables(self):
   batch_size = 5
   height, width = 224, 224
   num_classes = 1000
   with self.cached_session():
     inputs = random_ops.random_uniform((batch_size, height, width, 3))
     alexnet.alexnet_v2(inputs, num_classes)
     expected_names = [
         'alexnet_v2/conv1/weights',
         'alexnet_v2/conv1/biases',
         'alexnet_v2/conv2/weights',
         'alexnet_v2/conv2/biases',
         'alexnet_v2/conv3/weights',
         'alexnet_v2/conv3/biases',
         'alexnet_v2/conv4/weights',
         'alexnet_v2/conv4/biases',
         'alexnet_v2/conv5/weights',
         'alexnet_v2/conv5/biases',
         'alexnet_v2/fc6/weights',
         'alexnet_v2/fc6/biases',
         'alexnet_v2/fc7/weights',
         'alexnet_v2/fc7/biases',
         'alexnet_v2/fc8/weights',
         'alexnet_v2/fc8/biases',
     ]
     model_variables = [v.op.name for v in variables_lib.get_model_variables()]
     self.assertSetEqual(set(model_variables), set(expected_names))

示例#4

 def testModelVariables(self):
     batch_size = 5
     height, width = 224, 224
     num_classes = 1000
     with self.cached_session():
         inputs = random_ops.random_uniform((batch_size, height, width, 3))
         vgg.vgg_a(inputs, num_classes)
         expected_names = [
             'vgg_a/conv1/conv1_1/weights',
             'vgg_a/conv1/conv1_1/biases',
             'vgg_a/conv2/conv2_1/weights',
             'vgg_a/conv2/conv2_1/biases',
             'vgg_a/conv3/conv3_1/weights',
             'vgg_a/conv3/conv3_1/biases',
             'vgg_a/conv3/conv3_2/weights',
             'vgg_a/conv3/conv3_2/biases',
             'vgg_a/conv4/conv4_1/weights',
             'vgg_a/conv4/conv4_1/biases',
             'vgg_a/conv4/conv4_2/weights',
             'vgg_a/conv4/conv4_2/biases',
             'vgg_a/conv5/conv5_1/weights',
             'vgg_a/conv5/conv5_1/biases',
             'vgg_a/conv5/conv5_2/weights',
             'vgg_a/conv5/conv5_2/biases',
             'vgg_a/fc6/weights',
             'vgg_a/fc6/biases',
             'vgg_a/fc7/weights',
             'vgg_a/fc7/biases',
             'vgg_a/fc8/weights',
             'vgg_a/fc8/biases',
         ]
         model_variables = [
             v.op.name for v in variables_lib.get_model_variables()
         ]
         self.assertSetEqual(set(model_variables), set(expected_names))

示例#5

 def testModelVariables(self):
     batch_size = 5
     height, width = 224, 224
     num_classes = 1000
     with self.test_session():
         inputs = random_ops.random_uniform((batch_size, height, width, 3))
         alexnet.alexnet_v2(inputs, num_classes)
         expected_names = [
             'alexnet_v2/conv1/weights',
             'alexnet_v2/conv1/biases',
             'alexnet_v2/conv2/weights',
             'alexnet_v2/conv2/biases',
             'alexnet_v2/conv3/weights',
             'alexnet_v2/conv3/biases',
             'alexnet_v2/conv4/weights',
             'alexnet_v2/conv4/biases',
             'alexnet_v2/conv5/weights',
             'alexnet_v2/conv5/biases',
             'alexnet_v2/fc6/weights',
             'alexnet_v2/fc6/biases',
             'alexnet_v2/fc7/weights',
             'alexnet_v2/fc7/biases',
             'alexnet_v2/fc8/weights',
             'alexnet_v2/fc8/biases',
         ]
         model_variables = [
             v.op.name for v in variables_lib.get_model_variables()
         ]
         self.assertSetEqual(set(model_variables), set(expected_names))

示例#6

文件： overfeat_test.py 项目： yonghankim/tf-slim

 def testModelVariables(self):
     batch_size = 5
     height, width = 231, 231
     num_classes = 1000
     with self.cached_session():
         inputs = random_ops.random_uniform((batch_size, height, width, 3))
         overfeat.overfeat(inputs, num_classes)
         expected_names = [
             'overfeat/conv1/weights',
             'overfeat/conv1/biases',
             'overfeat/conv2/weights',
             'overfeat/conv2/biases',
             'overfeat/conv3/weights',
             'overfeat/conv3/biases',
             'overfeat/conv4/weights',
             'overfeat/conv4/biases',
             'overfeat/conv5/weights',
             'overfeat/conv5/biases',
             'overfeat/fc6/weights',
             'overfeat/fc6/biases',
             'overfeat/fc7/weights',
             'overfeat/fc7/biases',
             'overfeat/fc8/weights',
             'overfeat/fc8/biases',
         ]
         model_variables = [
             v.op.name for v in variables_lib.get_model_variables()
         ]
         self.assertSetEqual(set(model_variables), set(expected_names))

示例#7

文件： vgg_test.py 项目： 1000sprites/tensorflow

 def testModelVariables(self):
   batch_size = 5
   height, width = 224, 224
   num_classes = 1000
   with self.test_session():
     inputs = random_ops.random_uniform((batch_size, height, width, 3))
     vgg.vgg_a(inputs, num_classes)
     expected_names = [
         'vgg_a/conv1/conv1_1/weights',
         'vgg_a/conv1/conv1_1/biases',
         'vgg_a/conv2/conv2_1/weights',
         'vgg_a/conv2/conv2_1/biases',
         'vgg_a/conv3/conv3_1/weights',
         'vgg_a/conv3/conv3_1/biases',
         'vgg_a/conv3/conv3_2/weights',
         'vgg_a/conv3/conv3_2/biases',
         'vgg_a/conv4/conv4_1/weights',
         'vgg_a/conv4/conv4_1/biases',
         'vgg_a/conv4/conv4_2/weights',
         'vgg_a/conv4/conv4_2/biases',
         'vgg_a/conv5/conv5_1/weights',
         'vgg_a/conv5/conv5_1/biases',
         'vgg_a/conv5/conv5_2/weights',
         'vgg_a/conv5/conv5_2/biases',
         'vgg_a/fc6/weights',
         'vgg_a/fc6/biases',
         'vgg_a/fc7/weights',
         'vgg_a/fc7/biases',
         'vgg_a/fc8/weights',
         'vgg_a/fc8/biases',
     ]
     model_variables = [v.op.name for v in variables_lib.get_model_variables()]
     self.assertSetEqual(set(model_variables), set(expected_names))

示例#8

文件： variables_test.py 项目： AliMiraftab/tensorflow

 def testNameAndShape(self):
   with self.test_session():
     with variable_scope.variable_scope('A'):
       a = variables_lib2.model_variable('a', [5])
       self.assertEquals(a.op.name, 'A/a')
       self.assertListEqual(a.get_shape().as_list(), [5])
       self.assertListEqual([a], variables_lib2.get_model_variables('A'))

示例#9

文件： inception_v3_test.py 项目： 1000sprites/tensorflow

 def testModelHasExpectedNumberOfParameters(self):
   batch_size = 5
   height, width = 299, 299
   inputs = random_ops.random_uniform((batch_size, height, width, 3))
   with arg_scope(inception_v3.inception_v3_arg_scope()):
     inception_v3.inception_v3_base(inputs)
   total_params, _ = model_analyzer.analyze_vars(
       variables_lib.get_model_variables())
   self.assertAlmostEqual(21802784, total_params)

示例#10

 def testModelHasExpectedNumberOfParameters(self):
     batch_size = 5
     height, width = 224, 224
     inputs = random_ops.random_uniform((batch_size, height, width, 3))
     with arg_scope(inception_v2.inception_v2_arg_scope()):
         inception_v2.inception_v2_base(inputs)
     total_params, _ = model_analyzer.analyze_vars(
         variables_lib.get_model_variables())
     self.assertAlmostEqual(10173112, total_params)