注:对于图像分割而言,交叉熵的输入:通过深度学习模型的预测样本的输入为:SCHW,其中,S为样本数目,C为分类数目,H和W为图像样本尺寸。 样本标签的输入为:SH*W, 其中,S为样本数,H和W为图像样本尺寸。
参考链接:
https://blog.csdn.net/qq_22210253/article/details/85229988?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task
pytorch中交叉熵的妙用
最新推荐文章于 2023-04-17 14:00:00 发布