(self, x)
return landmarks
aux = self.conv1_aux(out1)
        aux = self.conv2_aux(aux)
        aux = self.conv3_aux(aux)
        aux = self.conv4_aux(aux)
        aux = self.max_pool1_aux(aux)
        aux = aux.view(aux.size(0), -1)
        aux = self.fc1_aux(aux)
        aux = torch.cat([aux, multi_scale], 1)
        pose = self.fc2_aux(aux)
        
        return pose, landmarks
aux = self.conv1_aux(out1)
        aux = self.conv2_aux(aux)
        aux = self.conv3_aux(aux)
        aux = self.conv4_aux(aux)
        aux = self.max_pool1_aux(aux)
        aux = aux.view(aux.size(0), -1)
        aux = self.fc1_aux(aux)
        aux = torch.cat([aux, multi_scale], 1)
        pose = self.fc2_aux(aux)
        
        return pose, landmarks
aux = self . conv1_aux ( out1 ) aux = self . conv2_aux ( aux ) aux = self . conv3_aux ( aux ) aux = self . conv4_aux ( aux ) aux = self . max_pool1_aux ( aux ) aux = aux . view ( aux . size ( 0 ) - 1 ) aux = self . fc1_aux ( aux ) aux = torch . cat ( [ aux multi_scale ] 1 ) pose = self . fc2_aux ( aux ) return pose landmarks
def forward(self, x):  # x: 3, 112, 112
        x = self.relu(self.bn1(self.conv1(x)))  # [64, 56, 56]
        #x = self.relu(self.bn2(self.conv2(x)))  # [64, 56, 56]
        x = self.relu(self.conv1_extra(self.dw_bn(self.dw_pool(x))))
        x = self.conv3_1(x)
        x = self.block3_2(x)
        x = self.block3_3(x)
        x = self.block3_4(x)
        out1 = self.block3_5(x)

        x = self.conv4_1(out1)
        x = self.conv5_1(x)
        x = self.block5_2(x)
        x = self.block5_3(x)
        x = self.block5_4(x)
        x = self.block5_5(x)
        x = self.block5_6(x)
        x = self.conv6_1(x)
        x1 = self.avg_pool1(x)
        x1 = x1.view(x1.size(0), -1)

        x = self.conv7(x)
        x2 = self.avg_pool2(x)
        x2 = x2.view(x2.size(0), -1)

        x3 = self.relu(self.conv8(x))
        x3 = x3.view(x1.size(0), -1)

        multi_scale = torch.cat([x1, x2, x3], 1)
        landmarks = self.fc(multi_scale)

        '''
        aux = self.conv1_aux(out1)
        aux = self.conv2_aux(aux)
        aux = self.conv3_aux(aux)
        aux = self.conv4_aux(aux)
        aux = self.max_pool1_aux(aux)
        aux = aux.view(aux.size(0), -1)
        aux = self.fc1_aux(aux)
        aux = torch.cat([aux, multi_scale], 1)
        pose = self.fc2_aux(aux)
        
        return pose, landmarks
        '''
        return landmarks
def forward ( self , x ) : # x: 3, 112, 112 x = self . relu ( self . bn1 ( self . conv1 ( x ) ) ) # [64, 56, 56] #x = self.relu(self.bn2(self.conv2(x)))  # [64, 56, 56] x = self . relu ( self . conv1_extra ( self . dw_bn ( self . dw_pool ( x ) ) ) ) x = self . conv3_1 ( x ) x = self . block3_2 ( x ) x = self . block3_3 ( x ) x = self . block3_4 ( x ) out1 = self . block3_5 ( x ) x = self . conv4_1 ( out1 ) x = self . conv5_1 ( x ) x = self . block5_2 ( x ) x = self . block5_3 ( x ) x = self . block5_4 ( x ) x = self . block5_5 ( x ) x = self . block5_6 ( x ) x = self . conv6_1 ( x ) x1 = self . avg_pool1 ( x ) x1 = x1 . view ( x1 . size ( 0 ) , - 1 ) x = self . conv7 ( x ) x2 = self . avg_pool2 ( x ) x2 = x2 . view ( x2 . size ( 0 ) , - 1 ) x3 = self . relu ( self . conv8 ( x ) ) x3 = x3 . view ( x1 . size ( 0 ) , - 1 ) multi_scale = torch . cat ( [ x1 , x2 , x3 ] , 1 ) landmarks = self . fc ( multi_scale ) return landmarks