将hrnet网络去掉last_layer层后连接胶囊网络构成新的网络，应该怎么在hrnet网络的HighResolutionModule结构代码中增加胶囊网络的PrimaryCapsLayer、AgreementRouting和CapsLayer的结构代码

要将HRNet网络的HighResolutionModule结构与胶囊网络的PrimaryCapsLayer、AgreementRouting和CapsLayer结构连接起来，需要进行如下步骤：

1. 首先，在HRNet的HighResolutionModule结构中增加胶囊网络的PrimaryCapsLayer层。在HRNet的代码中，可以在`hrnet.py`文件中找到HighResolutionModule类。在该类中的`_make_stage`函数中，可以添加胶囊网络的PrimaryCapsLayer层的代码。这里假设PrimaryCapsLayer层的输出通道数为`num_primary_caps`，代码示例如下：

# _make_stage函数中的代码
def _make_stage(self, layer_config, num_inchannels, multi_scale_output):
    ... # 其他代码
    # 添加胶囊网络的PrimaryCapsLayer层
    self.primary_caps = PrimaryCapsLayer(in_channels=num_inchannels, out_channels=num_primary_caps, num_routing=num_routing)
    ...

2. 接下来，在HRNet的HighResolutionModule结构中增加胶囊网络的AgreementRouting层。同样，在`hrnet.py`文件中找到HighResolutionModule类，在该类中的`forward`函数中，可以添加胶囊网络的AgreementRouting层的代码。这里假设AgreementRouting层的迭代次数为`num_iterations`，代码示例如下：

# forward函数中的代码
def forward(self, x):
    ... # 其他代码
    # 主要网络传递
    x = self.transition(x) + self.primary_caps(x)
    # 添加胶囊网络的AgreementRouting层
    x = self.agreement_routing(x, num_iterations=num_iterations)
    ...

3. 最后，在HRNet的HighResolutionModule结构中增加胶囊网络的CapsLayer层。同样，在`hrnet.py`文件中找到HighResolutionModule类，在该类中的`forward`函数中，可以添加胶囊网络的CapsLayer层的代码。这里假设CapsLayer层的输出通道数为`num_caps`，代码示例如下：

# forward函数中的代码
def forward(self, x):
    ... # 其他代码
    # 主要网络传递
    x = self.transition(x) + self.primary_caps(x)
    x = self.agreement_routing(x, num_iterations=num_iterations)
    # 添加胶囊网络的CapsLayer层
    x = self.caps_layer(x, num_caps=num_caps)
    ...

以上是一种将HRNet网络的HighResolutionModule结构与胶囊网络的PrimaryCapsLayer、AgreementRouting和CapsLayer结构连接起来的方法。根据具体情况，你可能需要根据你的网络架构和需要进行一些调整和修改。希望这些代码示例能给你提供一些帮助。