Open songyadong106 opened 4 years ago
你好，很高兴看到您的代码分享，有一个问题我想问一下，就是模型输出的时候为什么是[B,1,256,256,256]，这个1不应该是对应类别吗？ 假如说是三分类，模型输出不应该是[B,3,256,256,256]这个样子吗？ 希望可以解决一下我的困惑，谢谢！
你好，很高兴看到您的代码分享，有一个问题我想问一下，就是模型输出的时候为什么是[B,1,256,256,256]，这个1不应该是对应类别吗？ 假如说是三分类，模型输出不应该是[B,3,256,256,256]这个样子吗？ 希望可以解决一下我的困惑，谢谢！