Closed bigKoki closed 3 years ago
我自己写的yolov3的loss总是计算出nan值，所以我用您的loss计算函数放在我的代码上想看下是否是我的loss计算有问题。我发现在您的loss计算 cls loss时有这么一段： label_cls = label[..., 6:] label_mix = label[..., 5:6] 按道理，label cls应该是label[...,5:] 我想请问下您的第五位代表什么含义，因为最近事情很多，没时间细看您的dataloader部分，如果有时间的话麻烦简单讲解一下这部分处理在您的代码的哪部分，谢谢！！！
已經解決了，mix是在加載數據集時做的圖像融合
bigKoki
commented
3 years ago bigKoki
commented
3 years ago
我自己写的yolov3的loss总是计算出nan值，所以我用您的loss计算函数放在我的代码上想看下是否是我的loss计算有问题。我发现在您的loss计算 cls loss时有这么一段： label_cls = label[..., 6:] label_mix = label[..., 5:6] 按道理，label cls应该是label[...,5:] 我想请问下您的第五位代表什么含义，因为最近事情很多，没时间细看您的dataloader部分，如果有时间的话麻烦简单讲解一下这部分处理在您的代码的哪部分，谢谢！！！