def create_block_list(self):
self.clear_list()
backbone = self.factory.get_base_model(BackboneName.GhostNet,
self.data_channel)
base_out_channels = backbone.get_outchannel_list()
self.add_block_list(BlockType.BaseNet, backbone, base_out_channels[-1])
avgpool = nn.AdaptiveAvgPool2d((1, 1))
self.add_block_list(LayerType.GlobalAvgPool, avgpool,
base_out_channels[-1])
output_channel = 1280
layer1 = FcLayer(base_out_channels[-1], output_channel)
self.add_block_list(layer1.get_name(), layer1, output_channel)
layer2 = NormalizeLayer(bn_name=NormalizationType.BatchNormalize1d,
out_channel=output_channel)
self.add_block_list(layer2.get_name(), layer2, output_channel)
layer3 = ActivationLayer(self.activation_name, inplace=False)
self.add_block_list(layer3.get_name(), layer3, output_channel)
layer4 = nn.Dropout(0.2)
self.add_block_list(LayerType.Dropout, layer4, output_channel)
layer5 = nn.Linear(output_channel, self.class_number)
self.add_block_list(LayerType.FcLinear, layer5, self.class_number)
self.create_loss()
def create_block_list(self):
self.clear_list()
backbone = PointNet(data_channel=self.data_channel,
feature_transform=self.feature_transform,
bn_name=self.bn_name,
activation_name=self.activation_name)
base_out_channels = backbone.get_outchannel_list()
self.add_block_list(BlockType.BaseNet, backbone, base_out_channels[-1])
input_channel = self.block_out_channels[-1]
fc1 = FcBNActivationBlock(input_channel,
512,
bnName=self.bn_name,
activationName=self.activation_name)
self.add_block_list(fc1.get_name(), fc1, 512)
input_channel = 512
fc2 = nn.Linear(input_channel, 256)
self.add_block_list(LayerType.FcLinear, fc2, 256)
input_channel = 256
dropout = nn.Dropout(p=0.3)
self.add_block_list(LayerType.Dropout, dropout, input_channel)
normalize = NormalizeLayer(self.bn_name, input_channel)
self.add_block_list(normalize.get_name(), normalize, input_channel)
activate = ActivationLayer(self.activation_name, inplace=False)
self.add_block_list(activate.get_name(), activate, input_channel)
fc3 = nn.Linear(input_channel, self.class_number)
self.add_block_list(LayerType.FcLinear, fc3, self.class_number)
self.create_loss()
def create_block_list(self):
self.clear_list()
layer1 = ConvBNActivationBlock(in_channels=self.data_channel,
out_channels=self.num_init_features,
kernel_size=7,
stride=2,
padding=3,
bnName=self.bnName,
activationName=self.activationName)
self.add_block_list(layer1.get_name(), layer1, self.num_init_features)
layer2 = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
self.add_block_list(LayerType.MyMaxPool2d, layer2,
self.num_init_features)
self.in_channels = self.num_init_features
for index, num_block in enumerate(self.num_blocks):
self.make_densenet_layer(num_block, self.dilations[index],
self.bn_size, self.growth_rate,
self.drop_rate, self.bnName,
self.activationName)
self.in_channels = self.block_out_channels[-1]
if index != len(self.num_blocks) - 1:
trans = TransitionBlock(in_channel=self.in_channels,
output_channel=self.in_channels // 2,
stride=1,
bnName=self.bnName,
activationName=self.activationName)
self.add_block_list(trans.get_name(), trans,
self.in_channels // 2)
avg_pool = nn.AvgPool2d(kernel_size=2, stride=2)
self.add_block_list(LayerType.GlobalAvgPool, avg_pool,
self.block_out_channels[-1])
self.in_channels = self.block_out_channels[-1]
layer3 = NormalizeLayer(bn_name=self.bnName,
out_channel=self.in_channels)
self.add_block_list(layer3.get_name(), layer3, self.in_channels)
layer4 = ActivationLayer(self.activationName, False)
self.add_block_list(layer4.get_name(), layer4, self.in_channels)