How to get the stage4 features of hrnet?

[hiyyg]

It seems the hrnet currently can output features for all stages, but there is only one feature for each stage. However, I want to obtain the 4 features from all branches of stage 4. How can I achieve this goal?

[rwightman]

@hiyyg that is the way it (already) works for HRNet, with the exception of the stem (feature level 0), feature level 1-4 are the branch outputs of the last stage, stage4. The description may have been wrong somewhere as HRNet isn't like other nets. You can check the code.

        x = self.conv1(x)
        x = self.bn1(x)
        x = self.act1(x)
        if 0 in self._out_idx:
            out.append(x)
        x = self.conv2(x)
        x = self.bn2(x)
        x = self.act2(x)
        x = self.stages(x)
        if self.incre_modules is not None:
            x = [incre(f) for f, incre in zip(x, self.incre_modules)]
        for i, f in enumerate(x):
            if i + 1 in self._out_idx:
                out.append(f)

[hiyyg]

Thanks for clarification. I think it would be better if this could be documented or using another way to describe the features.