Python roi_pool_backward примеры использования

Пример #1

 def backward(ctx, grad_output):
     input, rois, argmax = ctx.saved_tensors
     output_size = ctx.output_size
     spatial_scale = ctx.spatial_scale
     bs, ch, h, w = ctx.input_shape
     grad_input = _C.roi_pool_backward(
         grad_output,
         input,
         rois,
         argmax,
         spatial_scale,
         output_size[0],
         output_size[1],
         bs,
         ch,
         h,
         w,
     )
     return grad_input, None, None, None

Пример #2

Файл: roi_pool.py Проект: laycoding/maskrcnn-benchmark

 def backward(ctx, grad_output):
     input, rois, argmax = ctx.saved_tensors
     output_size = ctx.output_size
     spatial_scale = ctx.spatial_scale
     bs, ch, h, w = ctx.input_shape
     grad_input = _C.roi_pool_backward(
         grad_output,
         input,
         rois,
         argmax,
         spatial_scale,
         output_size[0],
         output_size[1],
         bs,
         ch,
         h,
         w,
     )
     return grad_input, None, None, None