def forward(ctx, input, roi, output_size, spatial_scale, sampling_ratio): ctx.save_for_backward(roi) ctx.output_size = _pair(output_size) ctx.spatial_scale = spatial_scale ctx.sampling_ratio = sampling_ratio ctx.input_shape = input.size() output = _C.roi_align_forward( input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio ) return output
def forward(ctx, input, roi, output_size, spatial_scale, sampling_ratio): """ :param input: 某个level的feature map :param roi: 属于该level的roi :param output_size: 7 :param spatial_scale: (1/4, 1/8, 1/16, 1/32) or :param sampling_ratio: 2 """ ctx.save_for_backward(roi) # tuple(repeat(x, 2)) ctx.output_size = _pair(output_size) ctx.spatial_scale = spatial_scale ctx.sampling_ratio = sampling_ratio ctx.input_shape = input.size() output = _C.roi_align_forward(input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio) return output