摘要:        
lightweight openpose因为有3层stride=2的卷积层,所以他的标签(标签就是mask和keypoint还有paf)需要缩放3倍。 hrnet他是有设定的,输入大小和输出大小都是固定的。    阅读全文
posted @ 2023-02-05 21:27
祥瑞哈哈哈
阅读(43)
评论(0)
推荐(0)
        
            
        
        
摘要:        
我去将mask和lable缩小8倍,因为模型中有3层卷积层步数为2。 def __getitem__(self, idx): label = copy.deepcopy(self._labels[idx]) # label modified in transform image = cv2.imre    阅读全文
posted @ 2023-02-05 20:49
祥瑞哈哈哈
阅读(284)
评论(0)
推荐(0)
        
            
        
        
摘要:        
lightweight openpose的loss都用的是平方损失。 def l2_loss(input, target, mask, batch_size): loss = (input - target) * mask loss = (loss * loss) / 2 / batch_size     阅读全文
posted @ 2023-02-05 17:05
祥瑞哈哈哈
阅读(126)
评论(0)
推荐(0)
        
            
        
        
摘要:        
因为image对应一个mask,keypoint的mask和paf的mask都用的mask。 for idx in range(keypoint_mask.shape[0]): keypoint_mask[idx] = mask sample['keypoint_mask'] = keypoint_    阅读全文
posted @ 2023-02-05 10:45
祥瑞哈哈哈
阅读(18)
评论(0)
推荐(0)
        
 
                    
                
 浙公网安备 33010602011771号
浙公网安备 33010602011771号