|
| | __init__ (self, device="cpu", QP=32) |
| |
| | residual_unit_stg1 (self, x, nr) |
| | Generic residual unit.
|
| |
| | residual_unit_stg2 (self, x, nr) |
| | Generic residual unit.
|
| |
| | nr_calc (self, ac, ap) |
| | Calculate the number of residual units.
|
| |
| | split (self, cu, coords, sizes, split) |
| | Splits feature maps in specific way.
|
| |
| | forward (self, cu, sizes=None, coords=None) |
| | This functions propagates the input to the output.
|
| |
◆ __init__()
| msecnn_raulkviana.msecnn.MseCnnStg1.__init__ |
( |
|
self, |
|
|
|
device = "cpu", |
|
|
|
QP = 32 |
|
) |
| |
◆ forward()
| msecnn_raulkviana.msecnn.MseCnnStg1.forward |
( |
|
self, |
|
|
|
cu, |
|
|
|
sizes = None, |
|
|
|
coords = None |
|
) |
| |
This functions propagates the input to the output.
- Parameters
-
| [in] | cu | Input to the model |
| [out] | logits | Vector of raw predictions that a classification model generates |
Reimplemented in msecnn_raulkviana.msecnn.MseCnnStgX.
◆ nr_calc()
| msecnn_raulkviana.msecnn.MseCnnStg1.nr_calc |
( |
|
self, |
|
|
|
ac, |
|
|
|
ap |
|
) |
| |
Calculate the number of residual units.
- Parameters
-
| [in] | ac | Minimum value of the current input axises |
| [in] | ap | Minimum value of the parent input axises |
| [out] | nr | Number of residual units |
◆ residual_unit_stg1()
| msecnn_raulkviana.msecnn.MseCnnStg1.residual_unit_stg1 |
( |
|
self, |
|
|
|
x, |
|
|
|
nr |
|
) |
| |
◆ residual_unit_stg2()
| msecnn_raulkviana.msecnn.MseCnnStg1.residual_unit_stg2 |
( |
|
self, |
|
|
|
x, |
|
|
|
nr |
|
) |
| |
◆ split()
| msecnn_raulkviana.msecnn.MseCnnStg1.split |
( |
|
self, |
|
|
|
cu, |
|
|
|
coords, |
|
|
|
sizes, |
|
|
|
split |
|
) |
| |
Splits feature maps in specific way.
- Parameters
-
| [in] | cu | Input to the model |
| [in] | coords | Coordinates of the new CUs |
| [in] | size | Size of the new CUs |
| [in] | split | Way to split CU |
| [out] | cu_out | New Feature maps |
◆ activation_PRelu2_stg1
| msecnn_raulkviana.msecnn.MseCnnStg1.activation_PRelu2_stg1 |
◆ activation_PRelu2_stg2
| msecnn_raulkviana.msecnn.MseCnnStg1.activation_PRelu2_stg2 |
◆ activation_PRelu_stg1
| msecnn_raulkviana.msecnn.MseCnnStg1.activation_PRelu_stg1 |
◆ activation_PRelu_stg2
| msecnn_raulkviana.msecnn.MseCnnStg1.activation_PRelu_stg2 |
◆ first_simple_conv
| msecnn_raulkviana.msecnn.MseCnnStg1.first_simple_conv |
◆ simple_conv2_stg1
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv2_stg1 |
◆ simple_conv2_stg2
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv2_stg2 |
◆ simple_conv_no_activation2_stg1
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv_no_activation2_stg1 |
◆ simple_conv_no_activation2_stg2
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv_no_activation2_stg2 |
◆ simple_conv_no_activation_stg1
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv_no_activation_stg1 |
◆ simple_conv_no_activation_stg2
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv_no_activation_stg2 |
◆ simple_conv_stg1
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv_stg1 |
◆ simple_conv_stg2
| msecnn_raulkviana.msecnn.MseCnnStg1.simple_conv_stg2 |
◆ sub_net
| msecnn_raulkviana.msecnn.MseCnnStg1.sub_net |
The documentation for this class was generated from the following file: