Public Member Functions | Static Private Member Functions | Private Attributes
operator_py.proposal.ProposalOperator Class Reference

List of all members.

Public Member Functions

def __init__
def backward
def forward

Static Private Member Functions

def _clip_pad
def _filter_boxes

Private Attributes

 _anchors
 _feat_stride
 _num_anchors
 _output_score
 _ratios
 _rpn_min_size
 _rpn_post_nms_top_n
 _rpn_pre_nms_top_n
 _scales
 _threshold

Detailed Description

Definition at line 18 of file proposal.py.


Constructor & Destructor Documentation

def operator_py.proposal.ProposalOperator.__init__ (   self,
  feat_stride,
  scales,
  ratios,
  output_score,
  rpn_pre_nms_top_n,
  rpn_post_nms_top_n,
  threshold,
  rpn_min_size 
)

Definition at line 19 of file proposal.py.


Member Function Documentation

def operator_py.proposal.ProposalOperator._clip_pad (   tensor,
  pad_shape 
) [static, private]
Clip boxes of the pad area.
:param tensor: [n, c, H, W]
:param pad_shape: [h, w]
:return: [n, c, h, w]

Definition at line 171 of file proposal.py.

def operator_py.proposal.ProposalOperator._filter_boxes (   boxes,
  min_size 
) [static, private]
Remove all boxes with any side smaller than min_size 

Definition at line 163 of file proposal.py.

def operator_py.proposal.ProposalOperator.backward (   self,
  req,
  out_grad,
  in_data,
  out_data,
  in_grad,
  aux 
)

Definition at line 157 of file proposal.py.

def operator_py.proposal.ProposalOperator.forward (   self,
  is_train,
  req,
  in_data,
  out_data,
  aux 
)

Definition at line 38 of file proposal.py.


Member Data Documentation

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.

Definition at line 19 of file proposal.py.


The documentation for this class was generated from the following file:


rail_object_detector
Author(s):
autogenerated on Sat Jun 8 2019 20:26:31