def initialize(self, density_num_layers, density_channels, density_level, gcn_layers, gcn_channels, head_layernum, head_chn, upsample_layers, upsample_chn): self.backbone = hrnet.Body() self.upsample = UpSample(upsample_layers, upsample_chn) self.head = Head(head_layernum, head_chn) self.head2 = Head(head_layernum, head_chn) # self.head_density = Head(head_layernum, head_chn) # self.density_branch = DensityBranch(density_num_layers, density_channels, config.num_pts*2) # self.id_branch = DensityBranch(density_num_layers, density_channels, config.id_featdim * config.num_pts) # self.density_branch = M.ConvLayer(1, config.num_pts) self.c1 = M.ConvLayer(1, config.num_pts) self.c2 = M.ConvLayer(1, config.num_pts)
def initialize(self, head_layernum, head_chn, upsample_layers, upsample_chn): self.backbone = hrnet.Body() self.upsample = UpSample(upsample_layers, upsample_chn) self.head = Head(head_layernum, head_chn) self.c1 = M.ConvLayer(1, config.num_pts)