|
|
|
@ -13,40 +13,71 @@
|
|
|
|
|
# limitations under the License.
|
|
|
|
|
|
|
|
|
|
# TODO: define specitial functions used in computer vision task
|
|
|
|
|
# __all__ = ['affine_channel',
|
|
|
|
|
# 'affine_grid',
|
|
|
|
|
# 'anchor_generator',
|
|
|
|
|
# 'bipartite_match',
|
|
|
|
|
# 'box_clip',
|
|
|
|
|
# 'box_coder',
|
|
|
|
|
# 'box_decoder_and_assign',
|
|
|
|
|
# 'collect_fpn_proposals',
|
|
|
|
|
# 'deformable_conv',
|
|
|
|
|
# 'deformable_roi_pooling',
|
|
|
|
|
# 'density_prior_box',
|
|
|
|
|
# 'detection_output',
|
|
|
|
|
# 'distribute_fpn_proposals',
|
|
|
|
|
# 'fsp_matrix',
|
|
|
|
|
# 'generate_mask_labels',
|
|
|
|
|
# 'generate_proposal_labels',
|
|
|
|
|
# 'generate_proposals',
|
|
|
|
|
# 'grid_sampler',
|
|
|
|
|
# 'image_resize',
|
|
|
|
|
# 'image_resize_short',
|
|
|
|
|
# 'multi_box_head',
|
|
|
|
|
# 'pixel_shuffle',
|
|
|
|
|
# 'prior_box',
|
|
|
|
|
# 'prroi_pool',
|
|
|
|
|
# 'psroi_pool',
|
|
|
|
|
# 'resize_bilinear',
|
|
|
|
|
# 'resize_nearest',
|
|
|
|
|
# 'resize_trilinear',
|
|
|
|
|
# 'retinanet_detection_output',
|
|
|
|
|
# 'retinanet_target_assign',
|
|
|
|
|
# 'roi_align',
|
|
|
|
|
# 'roi_perspective_transform',
|
|
|
|
|
# 'roi_pool',
|
|
|
|
|
# 'shuffle_channel',
|
|
|
|
|
# 'space_to_depth',
|
|
|
|
|
# 'yolo_box',
|
|
|
|
|
# 'yolov3_loss']
|
|
|
|
|
from ...fluid.layers import affine_channel #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import affine_grid #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import anchor_generator #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import bipartite_match #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import box_clip #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import box_coder #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import box_decoder_and_assign #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import collect_fpn_proposals #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import deformable_roi_pooling #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import density_prior_box #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import detection_output #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import distribute_fpn_proposals #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import generate_mask_labels #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import generate_proposal_labels #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import generate_proposals #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import grid_sampler #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import image_resize #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import prior_box #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import prroi_pool #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import psroi_pool #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import resize_bilinear #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import resize_nearest #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import resize_trilinear #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import roi_align #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import roi_pool #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import space_to_depth #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import yolo_box #DEFINE_ALIAS
|
|
|
|
|
from ...fluid.layers import yolov3_loss #DEFINE_ALIAS
|
|
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
|
'affine_channel',
|
|
|
|
|
'affine_grid',
|
|
|
|
|
'anchor_generator',
|
|
|
|
|
'bipartite_match',
|
|
|
|
|
'box_clip',
|
|
|
|
|
'box_coder',
|
|
|
|
|
'box_decoder_and_assign',
|
|
|
|
|
'collect_fpn_proposals',
|
|
|
|
|
# 'deformable_conv',
|
|
|
|
|
'deformable_roi_pooling',
|
|
|
|
|
'density_prior_box',
|
|
|
|
|
'detection_output',
|
|
|
|
|
'distribute_fpn_proposals',
|
|
|
|
|
# 'fsp_matrix',
|
|
|
|
|
'generate_mask_labels',
|
|
|
|
|
'generate_proposal_labels',
|
|
|
|
|
'generate_proposals',
|
|
|
|
|
'grid_sampler',
|
|
|
|
|
'image_resize',
|
|
|
|
|
# 'image_resize_short',
|
|
|
|
|
# 'multi_box_head',
|
|
|
|
|
# 'pixel_shuffle',
|
|
|
|
|
'prior_box',
|
|
|
|
|
'prroi_pool',
|
|
|
|
|
'psroi_pool',
|
|
|
|
|
'resize_bilinear',
|
|
|
|
|
'resize_nearest',
|
|
|
|
|
'resize_trilinear',
|
|
|
|
|
# 'retinanet_detection_output',
|
|
|
|
|
# 'retinanet_target_assign',
|
|
|
|
|
'roi_align',
|
|
|
|
|
# 'roi_perspective_transform',
|
|
|
|
|
'roi_pool',
|
|
|
|
|
# 'shuffle_channel',
|
|
|
|
|
'space_to_depth',
|
|
|
|
|
'yolo_box',
|
|
|
|
|
'yolov3_loss'
|
|
|
|
|
]
|
|
|
|
|