resnet3d.Bottleneck3d Class Reference

Inheritance diagram for resnet3d.Bottleneck3d:

Public Member Functions
None	__init__ (self, int inplanes, int planes, int spatial_stride=1, int temporal_stride=1, int dilation=1, Optional[nn.Module] downsample=None, bool inflate=True, str inflate_style='3x1x1')
	forward (self, x)

Public Attributes
	inplanes
	planes
	spatial_stride
	temporal_stride
	dilation
	inflate
	inflate_style
	conv1_stride_s
	conv2_stride_s
	conv1_stride_t
	conv2_stride_t
	conv1
	conv2
	conv3
	expansion
	downsample
	relu

Static Public Attributes
int	expansion = 4

Detailed Description

Definition at line 7 of file resnet3d.py.

Constructor & Destructor Documentation

__init__()

None resnet3d.Bottleneck3d.__init__	(		self,
		int	inplanes,
		int	planes,
		int	spatial_stride = 1,
		int	temporal_stride = 1,
		int	dilation = 1,
		Optional[nn.Module]	downsample = None,
		bool	inflate = True,
		str	inflate_style = '3x1x1' )

Definition at line 11 of file resnet3d.py.

                 inflate_style: str = '3x1x1') -> None:
        super().__init__()
 
        self.inplanes = inplanes
        self.planes = planes
        self.spatial_stride = spatial_stride
        self.temporal_stride = temporal_stride
        self.dilation = dilation
        
        self.inflate = inflate
        self.inflate_style = inflate_style
        
        self.conv1_stride_s = 1
        self.conv2_stride_s = spatial_stride
        self.conv1_stride_t = 1
        self.conv2_stride_t = temporal_stride
        
        conv1_stride = (self.conv1_stride_t, self.conv1_stride_s, self.conv1_stride_s)
        conv2_stride = (self.conv2_stride_t, self.conv2_stride_s, self.conv2_stride_s)
 
        if self.inflate:
            if inflate_style == '3x1x1':
                conv1_kernel_size = (3, 1, 1)
                conv1_padding = (1, 0, 0)
                conv2_kernel_size = (1, 3, 3)
                conv2_padding = (0, dilation, dilation)
            else:
                conv1_kernel_size = (1, 1, 1)
                conv1_padding = (0, 0, 0)
                conv2_kernel_size = (3, 3, 3)
                conv2_padding = (1, dilation, dilation)
        else:
            conv1_kernel_size = (1, 1, 1)
            conv1_padding = (0, 0, 0)
            conv2_kernel_size = (1, 3, 3)
            conv2_padding = (0, dilation, dilation)
 
 
        self.conv1 = nn.Sequential(nn.Conv3d(inplanes, planes, kernel_size=conv1_kernel_size, stride=conv1_stride, padding=conv1_padding, bias=False),
                                   nn.BatchNorm3d(planes), 
                                   nn.ReLU(inplace=True))
 
        self.conv2 = nn.Sequential(nn.Conv3d(planes, planes, kernel_size=conv2_kernel_size, stride=conv2_stride, padding=conv2_padding, bias=False),
                                   nn.BatchNorm3d(planes), 
                                   nn.ReLU(inplace=True))
 
        self.conv3 = nn.Sequential(nn.Conv3d(planes, planes * self.expansion, kernel_size=1, bias=False),
                                   nn.BatchNorm3d(planes * self.expansion))
 
        self.downsample = downsample
        self.relu = nn.ReLU(inplace=True)
 
 

Member Function Documentation

forward()

resnet3d.Bottleneck3d.forward	(		self,
			x )

Definition at line 72 of file resnet3d.py.

    def forward(self, x):
        
        identity = x
 
        out = self.conv1(x)
        out = self.conv2(out)
        out = self.conv3(out)
 
        if self.downsample is not None:
            identity = self.downsample(x)
 
        out = out + identity
        
        out = self.relu(out)
 
        return out
 
 

Member Data Documentation

conv1

resnet3d.Bottleneck3d.conv1

Definition at line 57 of file resnet3d.py.

conv1_stride_s

resnet3d.Bottleneck3d.conv1_stride_s

Definition at line 31 of file resnet3d.py.

conv1_stride_t

resnet3d.Bottleneck3d.conv1_stride_t

Definition at line 33 of file resnet3d.py.

conv2

resnet3d.Bottleneck3d.conv2

Definition at line 61 of file resnet3d.py.

conv2_stride_s

resnet3d.Bottleneck3d.conv2_stride_s

Definition at line 32 of file resnet3d.py.

conv2_stride_t

resnet3d.Bottleneck3d.conv2_stride_t

Definition at line 34 of file resnet3d.py.

conv3

resnet3d.Bottleneck3d.conv3

Definition at line 65 of file resnet3d.py.

dilation

resnet3d.Bottleneck3d.dilation

Definition at line 26 of file resnet3d.py.

downsample

resnet3d.Bottleneck3d.downsample

Definition at line 68 of file resnet3d.py.

expansion [1/2]

int resnet3d.Bottleneck3d.expansion = 4

static

Definition at line 9 of file resnet3d.py.

expansion [2/2]

resnet3d.Bottleneck3d.expansion

Definition at line 66 of file resnet3d.py.

inflate

resnet3d.Bottleneck3d.inflate

Definition at line 28 of file resnet3d.py.

inflate_style

resnet3d.Bottleneck3d.inflate_style

Definition at line 29 of file resnet3d.py.

inplanes

resnet3d.Bottleneck3d.inplanes

Definition at line 22 of file resnet3d.py.

planes

resnet3d.Bottleneck3d.planes

Definition at line 23 of file resnet3d.py.

relu

resnet3d.Bottleneck3d.relu

Definition at line 69 of file resnet3d.py.

spatial_stride

resnet3d.Bottleneck3d.spatial_stride

Definition at line 24 of file resnet3d.py.

temporal_stride

resnet3d.Bottleneck3d.temporal_stride

Definition at line 25 of file resnet3d.py.

---

The documentation for this class was generated from the following file:

• smaction/models/backbones/resnet3d.py