pyanomaly/networks/auxiliary/pose/models.py
def forward(self, x):
        """Forward function.
        The output 'x' is the 'heatmaps' representing the location of points.
        """
        x = self.conv1(x)
        x = self.bn1(x)
        x = self.relu(x)
        x = self.maxpool(x)

        x = self.layer1(x)
        x = self.layer2(x)
        x = self.layer3(x)
        x = self.layer4(x)

        x = self.deconv_layers(x)
        x = self.final_layer(x)
        
        # heatmaps = x
        # coords = x
        # if self.extra.TARGET_TYPE == 'points':
        #     if self.extra.USE_DSNT:
        #         heatmaps = dsntnn.flat_softmax(x)
        #         coords = dsntnn.dsnt(x)
        #     else:
        #         raise Exception('Not support other coords mapping methods')
        # else:
        #     heatmaps = x
        #     coords = x
        # return x, heatmaps, coords
        # return {'123':x, '456':x, '789':x}
        # return x, x, x
        # return x 
        return {'heatmaps':x}
Forward function.
        The output 'x' is the 'heatmaps' representing the location of points.
        
Forward function.
The output 'x' is the 'heatmaps' representing the location of points.
Forward function . The output ' x ' is the ' heatmaps ' representing the location of points .
def forward(self, x):
        x = self.conv1(x)
        x = self.bn1(x)
        x = self.relu(x)
        x = self.maxpool(x)
        x = self.layer1(x)
        x = self.layer2(x)
        x = self.layer3(x)
        x = self.layer4(x)
        x = self.deconv_layers(x)
        x = self.final_layer(x)
        return {'heatmaps':x}
def forward ( self , x ) : x = self . conv1 ( x ) x = self . bn1 ( x ) x = self . relu ( x ) x = self . maxpool ( x ) x = self . layer1 ( x ) x = self . layer2 ( x ) x = self . layer3 ( x ) x = self . layer4 ( x ) x = self . deconv_layers ( x ) x = self . final_layer ( x ) return { 'heatmaps' : x }
Forward function.
Forward function .
"""Forward function.
        The output 'x' is the 'heatmaps' representing the location of points.
        """ # heatmaps = x
 # coords = x
 # if self.extra.TARGET_TYPE == 'points':
 #     if self.extra.USE_DSNT:
 #         heatmaps = dsntnn.flat_softmax(x)
 #         coords = dsntnn.dsnt(x)
 #     else:
 #         raise Exception('Not support other coords mapping methods')
 # else:
 #     heatmaps = x
 #     coords = x
 # return x, heatmaps, coords
 # return {'123':x, '456':x, '789':x}
 # return x, x, x
 # return x 

self
x