Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@skrish13

skrish13/setup_layers.py

Created March 20, 2018 12:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save skrish13/e9bc482f18708ae10e5d9511fbae302b to your computer and use it in GitHub Desktop.
Save skrish13/e9bc482f18708ae10e5d9511fbae302b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
setup_layers.py
# Setup layers
self.mrcnn_mask_conv1 = nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1)
self.mrcnn_mask_bn1 = nn.BatchNorm2d(256, eps=0.001)
self.mrcnn_mask_conv2 = nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1)
self.mrcnn_mask_bn2 = nn.BatchNorm2d(256, eps=0.001)
self.mrcnn_mask_conv3 = nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1)
self.mrcnn_mask_bn3 = nn.BatchNorm2d(256, eps=0.001)
self.mrcnn_mask_conv4 = nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1)
self.mrcnn_mask_bn4 = nn.BatchNorm2d(256, eps=0.001)
self.mrcnn_mask_deconv = nn.ConvTranspose2d(256, 256, kernel_size=2, stride=2)
self.mrcnn_mask = nn.Conv2d(256, self.num_classes, kernel_size=1, stride=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment