maskrcnn_benchmark学习笔记——建模\_roi_heads_关键点头_inference.py

 import torch

    
 from torch import nn
    
  
    
  
    
 class KeypointPostProcessor(nn.Module):
    
     def __init__(self, keypointer=None):
    
     super(KeypointPostProcessor, self).__init__()
    
     self.keypointer = keypointer # xy_preds（x，1，y）, end_scores
    
  
    
     def forward(self, x, boxes):
    
     mask_prob = x
    
  
    
     scores = None
    
     if self.keypointer:
    
         mask_prob, scores = self.keypointer(x, boxes)  # mask_prob和 boxes →→ mask_prob, scores
    
  
    
     assert len(boxes) == 1, "Only non-batched inference supported for no