Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add octconv layers and obj coord preprocessing ops. #975

Open
wants to merge 20 commits into
base: master
Choose a base branch
from
Open
2 changes: 1 addition & 1 deletion docs/modules/cost.rst
Original file line number Diff line number Diff line change
Expand Up @@ -96,5 +96,5 @@ Special
.. autofunction:: maxnorm_i_regularizer

Huber Loss
^^^^^^^^^^
--------------------------
.. autofunction:: huber_loss
34 changes: 34 additions & 0 deletions docs/modules/layers.rst
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,12 @@ Layer list
SeparableConv2d
DeformableConv2d
GroupConv2d
OctConv2dIn
OctConv2d
OctConv2dOut
OctConv2dHighOut
OctConv2dLowOut
OctConv2dConcat

PadLayer
PoolLayer
Expand Down Expand Up @@ -241,6 +247,34 @@ GroupConv2d
"""""""""""""""""""""
.. autoclass:: GroupConv2d

OctConv
^^^^^^^^^^^^^^^^^^^^^^^^^^

For OctConv2d, see `Drop an Octave: Reducing Spatial Redundancy in Convolutional Neural Networks with Octave Convolution <https://arxiv.org/abs/1904.05049>`__.

OctConv2dIn
"""""""""""""""""""""
.. autoclass:: OctConv2dIn

OctConv2d
"""""""""""""""""""""
.. autoclass:: OctConv2d

OctConv2dOut
"""""""""""""""""""""
.. autoclass:: OctConv2dOut

OctConv2dHighOut
"""""""""""""""""""""
.. autoclass:: OctConv2dHighOut

OctConv2dConcat
"""""""""""""""""""""
.. autoclass:: OctConv2dConcat

OctConv2dLowOut
"""""""""""""""""""""
.. autoclass:: OctConv2dLowOut

Separable Convolutions
^^^^^^^^^^^^^^^^^^^^^^^^^^
Expand Down
10 changes: 10 additions & 0 deletions docs/modules/prepro.rst
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,8 @@ API - Data Pre-Processing
obj_box_coord_upleft_butright_to_centroid
obj_box_coord_centroid_to_upleft
obj_box_coord_upleft_to_centroid
obj_box_coord_affine
rotated_obj_box_coord_affine

parse_darknet_ann_str_to_list
parse_darknet_ann_list_to_cls_box
Expand Down Expand Up @@ -577,6 +579,14 @@ Image Aug - Zoom
^^^^^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: obj_box_zoom

Image Aug - Affine
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: obj_box_coord_affine

Image Aug - Rotated-Affine
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: rotated_obj_box_coord_affine

Keypoints
------------

Expand Down
9 changes: 9 additions & 0 deletions tensorlayer/layers/convolution/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
from .ternary_conv import *
from .quan_conv import *
from .quan_conv_bn import *
from .oct_conv import *

__all__ = [

Expand Down Expand Up @@ -80,4 +81,12 @@
#quan_conv
'QuanConv2d',
'QuanConv2dWithBN',

# octave_conv
'OctConv2dIn',
'OctConv2d',
'OctConv2dOut',
'OctConv2dHighOut',
'OctConv2dLowOut',
'OctConv2dConcat',
]
Loading