Advertisement

maskrcnn_benchmark学习笔记——modeling.roi_heads.keypoint_head.loss.py

阅读量:
复制代码
复制代码
 import torch

    
 from torch.nn import functional as F
    
  
    
 from maskrcnn_benchmark.modeling.matcher import Matcher
    
  
    
 from maskrcnn_benchmark.modeling.balanced_positive_negative_sampler import (
    
     BalancedPositiveNegativeSampler,
    
 )
    
 from maskrcnn_benchmark.structures.boxlist_ops import boxlist_iou
    
 from maskrcnn_benchmark.modeling.utils import cat
    
 from maskrcnn_benchmark.layers import smooth_l1_loss
    
 from maskrcnn_benchmark.structures.boxlist_ops import cat_boxlist
    
  
    
 from maskrcnn_benchmark.structures.keypoint 

全部评论 (0)

还没有任何评论哟~