Public Member Functions | |
def | __init__ |
def | backward |
def | forward |
Private Attributes | |
_num_classes | |
_num_reg_classes | |
_roi_per_img |
Definition at line 19 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMOperator.__init__ | ( | self, | |
num_classes, | |||
num_reg_classes, | |||
roi_per_img | |||
) |
Definition at line 20 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMOperator.backward | ( | self, | |
req, | |||
out_grad, | |||
in_data, | |||
out_data, | |||
in_grad, | |||
aux | |||
) |
Definition at line 56 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMOperator.forward | ( | self, | |
is_train, | |||
req, | |||
in_data, | |||
out_data, | |||
aux | |||
) |
Definition at line 26 of file box_annotator_ohem.py.
Definition at line 20 of file box_annotator_ohem.py.
Definition at line 20 of file box_annotator_ohem.py.
Definition at line 20 of file box_annotator_ohem.py.