pytorch-cpn的refine_loss
refine_output和 target7 的shape均为[batch,17,64,48],经过计算每个元素对应位置的loss后,得到[batch,17,64,48]的refine_loss,然后经过refine_loss.mean(dim=3).mean(dim=2),得到[batch,17]的loss。
target7 = torch.zeros((1,num_class, output_shape[0], output_shape[1]))
refine_output = torc
原创
2020-07-18 17:08:52 ·
480 阅读 ·
0 评论