paddle/fluid
operators
detection
collect_fpn_proposals_op.cc +25 -1
collect_fpn_proposals_op.cu +25 -5
collect_fpn_proposals_op.h +43 -6
distribute_fpn_proposals_op.cc +31 -1
distribute_fpn_proposals_op.cu +20 -5
distribute_fpn_proposals_op.h +42 -6
generate_proposals_op.cc +33 -9
generate_proposals_op.cu +18 -9
roi_align_op.cc +17 -9
roi_align_op.cu +21 -17
roi_align_op.h +17 -13
roi_pool_op.cc +17 -7
roi_pool_op.cu +22 -19
roi_pool_op.h +17 -13
pybind
op_function_generator.cc +9 -0
python/paddle/fluid
layers
detection.py +101 -23
nn.py +46 -17
tests
unittests
test_collect_fpn_proposals_op.py +33 -2
test_distribute_fpn_proposals_op.py +31 -1
test_generate_proposals_op.py +23 -10
test_layers.py +44 -17
test_roi_align_op.py +1 -6
test_roi_pool_op.py +1 -6
test_detection.py +217 -62