本文整理汇总了Python中nets.nets_factory.networks_map方法的典型用法代码示例。如果您正苦于以下问题：Python nets_factory.networks_map方法的具体用法？Python nets_factory.networks_map怎么用？Python nets_factory.networks_map使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类nets.nets_factory的用法示例。
在下文中一共展示了nets_factory.networks_map方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: testGetNetworkFn
# 需要导入模块: from nets import nets_factory [as 别名]
# 或者: from nets.nets_factory import networks_map [as 别名]
def testGetNetworkFn(self):
batch_size = 5
num_classes = 1000
for net in nets_factory.networks_map:
with self.test_session():
net_fn = nets_factory.get_network_fn(net, num_classes)
# Most networks use 224 as their default_image_size
image_size = getattr(net_fn, 'default_image_size', 224)
inputs = tf.random_uniform((batch_size, image_size, image_size, 3))
logits, end_points = net_fn(inputs)
self.assertTrue(isinstance(logits, tf.Tensor))
self.assertTrue(isinstance(end_points, dict))
self.assertEqual(logits.get_shape().as_list()[0], batch_size)
self.assertEqual(logits.get_shape().as_list()[-1], num_classes)