本文整理汇总了Python中tensorflow.python.ops.gen_image_ops.crop_and_resize_grad_boxes方法的典型用法代码示例。如果您正苦于以下问题:Python gen_image_ops.crop_and_resize_grad_boxes方法的具体用法?Python gen_image_ops.crop_and_resize_grad_boxes怎么用?Python gen_image_ops.crop_and_resize_grad_boxes使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.python.ops.gen_image_ops
的用法示例。
在下文中一共展示了gen_image_ops.crop_and_resize_grad_boxes方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: _CropAndResizeGrad
# 需要导入模块: from tensorflow.python.ops import gen_image_ops [as 别名]
# 或者: from tensorflow.python.ops.gen_image_ops import crop_and_resize_grad_boxes [as 别名]
def _CropAndResizeGrad(op, grad):
"""The derivatives for crop_and_resize.
We back-propagate to the image only when the input image tensor has floating
point dtype but we always back-propagate to the input boxes tensor.
Args:
op: The CropAndResize op.
grad: The tensor representing the gradient w.r.t. the output.
Returns:
The gradients w.r.t. the input image, boxes, as well as the always-None
gradients w.r.t. box_ind and crop_size.
"""
image = op.inputs[0]
if image.get_shape().is_fully_defined():
image_shape = image.get_shape().as_list()
else:
image_shape = array_ops.shape(image)
allowed_types = [dtypes.float16, dtypes.float32, dtypes.float64]
if op.inputs[0].dtype in allowed_types:
# pylint: disable=protected-access
grad0 = gen_image_ops.crop_and_resize_grad_image(grad,
op.inputs[1],
op.inputs[2],
image_shape,
T=op.get_attr("T"))
# pylint: enable=protected-access
else:
grad0 = None
grad1 = gen_image_ops.crop_and_resize_grad_boxes(grad, op.inputs[0],
op.inputs[1], op.inputs[2])
return [grad0, grad1, None, None]