Crate caffe2op_roialign

Structs

• BilinearInterpolationParam
• GetRoIAlignGradient
• GetRoIAlignRotatedGradient
• PreCalc
• RoIAlignGradientOp
• RoIAlignOp
  | Region of Interest (RoI) align operation | as used in Mask R-CNN. |
• RoIAlignRotatedGradientOp
• RoIAlignRotatedOp
  | Similar to RoIAlign but can handle rotated | region proposals. | | Based on https://arxiv.org/abs/1703.01086. |
• TestParams

Traits

• GetDeviceType

Functions

• add
• add_const_input
• add_input
• add_input_cpu_context
• add_input_cudacontext
• bilinear_interpolate_gradient
• create_and_run
• make_bilinear_interpolation_params
• pre_calc_for_bilinear_interpolate
• rand_int
• roi_align_backward_feature
• roi_align_rotated_forward

Type Definitions

• RoIAlignCPUOp
• RoIAlignGradientCPUOp
• RoIAlignRotatedOpFloatCPU