def METHOD_NAME(self, i, x):
    confidence = self.classification_headers[i](x)
    confidence = confidence.permute(0, 2, 3, 1).contiguous()
    confidence = confidence.view(confidence.size(0), -1, self.num_classes)
    location = self.regression_headers[i](x)
    location = location.permute(0, 2, 3, 1).contiguous()
    location = location.view(location.size(0), -1, 4)
    return confidence, location