本文整理汇总了Python中object_detection.utils.static_shape.get_batch_size方法的典型用法代码示例。如果您正苦于以下问题：Python static_shape.get_batch_size方法的具体用法？Python static_shape.get_batch_size怎么用？Python static_shape.get_batch_size使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类object_detection.utils.static_shape的用法示例。
在下文中一共展示了static_shape.get_batch_size方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_return_correct_batchSize
# 需要导入模块: from object_detection.utils import static_shape [as 别名]
# 或者: from object_detection.utils.static_shape import get_batch_size [as 别名]
def test_return_correct_batchSize(self):
tensor_shape = tf.TensorShape(dims=[32, 299, 384, 3])
self.assertEqual(32, static_shape.get_batch_size(tensor_shape)) 示例2: test_die_on_tensor_shape_with_rank_three
# 需要导入模块: from object_detection.utils import static_shape [as 别名]
# 或者: from object_detection.utils.static_shape import get_batch_size [as 别名]
def test_die_on_tensor_shape_with_rank_three(self):
tensor_shape = tf.TensorShape(dims=[32, 299, 384])
with self.assertRaises(ValueError):
static_shape.get_batch_size(tensor_shape)
static_shape.get_height(tensor_shape)
static_shape.get_width(tensor_shape)
static_shape.get_depth(tensor_shape)