Skip to content

Commit 71c1893

Browse files
committed
Add octconv layers and obj coord preprocessing ops.
Add octconv layers and obj coord preprocessing ops.
1 parent f88f23e commit 71c1893

File tree

7 files changed

+803
-1
lines changed

7 files changed

+803
-1
lines changed

docs/modules/cost.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -96,5 +96,5 @@ Special
9696
.. autofunction:: maxnorm_i_regularizer
9797

9898
Huber Loss
99-
^^^^^^^^^^
99+
--------------------------
100100
.. autofunction:: huber_loss

docs/modules/layers.rst

+34
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,12 @@ Layer list
3939
SeparableConv2d
4040
DeformableConv2d
4141
GroupConv2d
42+
OctConv2dIn
43+
OctConv2d
44+
OctConv2dOut
45+
OctConv2dHighOut
46+
OctConv2dLowOut
47+
OctConv2dConcat
4248

4349
PadLayer
4450
PoolLayer
@@ -241,6 +247,34 @@ GroupConv2d
241247
"""""""""""""""""""""
242248
.. autoclass:: GroupConv2d
243249

250+
OctConv2d
251+
--------------------------
252+
253+
For OctConv2d, see `Drop an Octave: Reducing Spatial Redundancy in Convolutional Neural Networks with Octave Convolution <https://arxiv.org/abs/1904.05049>`__.
254+
255+
OctConv2dIn
256+
"""""""""""""""""""""
257+
.. autoclass:: OctConv2dIn
258+
259+
OctConv2d
260+
"""""""""""""""""""""
261+
.. autoclass:: OctConv2d
262+
263+
OctConv2dOut
264+
"""""""""""""""""""""
265+
.. autoclass:: OctConv2dOut
266+
267+
OctConv2dHighOut
268+
"""""""""""""""""""""
269+
.. autoclass:: OctConv2dHighOut
270+
271+
OctConv2dConcat
272+
"""""""""""""""""""""
273+
.. autoclass:: OctConv2dConcat
274+
275+
OctConv2dLowOut
276+
"""""""""""""""""""""
277+
.. autoclass:: OctConv2dLowOut
244278

245279
Separable Convolutions
246280
^^^^^^^^^^^^^^^^^^^^^^^^^^

docs/modules/prepro.rst

+10
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,8 @@ API - Data Pre-Processing
7979
obj_box_coord_upleft_butright_to_centroid
8080
obj_box_coord_centroid_to_upleft
8181
obj_box_coord_upleft_to_centroid
82+
obj_box_coord_affine
83+
rotated_obj_box_coord_affine
8284

8385
parse_darknet_ann_str_to_list
8486
parse_darknet_ann_list_to_cls_box
@@ -577,6 +579,14 @@ Image Aug - Zoom
577579
^^^^^^^^^^^^^^^^^^^^^^^^^
578580
.. autofunction:: obj_box_zoom
579581

582+
Image Aug - Affine
583+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
584+
.. autofunction:: obj_box_coord_affine
585+
586+
Image Aug - Rotated-Affine
587+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
588+
.. autofunction:: rotated_obj_box_coord_affine
589+
580590
Keypoints
581591
------------
582592

tensorlayer/layers/convolution/__init__.py

+8
Original file line numberDiff line numberDiff line change
@@ -80,4 +80,12 @@
8080
#quan_conv
8181
'QuanConv2d',
8282
'QuanConv2dWithBN',
83+
84+
# octave_conv
85+
'OctConv2dIn',
86+
'OctConv2d',
87+
'OctConv2dOut',
88+
'OctConv2dHighOut',
89+
'OctConv2dLowOut',
90+
'OctConv2dConcat',
8391
]

0 commit comments

Comments
 (0)