我是 deep_learning 的初學者，我的畢業論文是關于語意分割的。但是我在使用庫的“segmentation-models”
我不明白（因為我的英語有限或缺乏知識......），并且 x_val 和 y_val 都有 3 個通道，它們是 numpy 陣列：
請幫我！！！非常感謝！
解決這個錯誤并訓練我的模型。
uj5u.com熱心網友回復：
看起來你有 3 個通道，但輸出的形狀只有 1 個通道。
UNET 初始化有一個名為 的引數classes，它定義了輸出形狀。您可能應該classes=3在定義模型時將其設定為。請參閱https://segmentation-models.readthedocs.io/en/latest/api.html#unet
中的
檔案，
更具體地說，請注意以下代碼段：
類 - 輸出類的數量（輸出形狀 - (h, w, classes)）。
澄清：輸入通道的數量在這里沒有實際意義。通道數在卷積層、跳過連接連接等中多次變化，因此輸出通道數與標簽通道數是在這種情況下唯一重要的事情。train_x可能有 1 個通道、37 個通道或任何其他必要數量的通道，并且您得到的錯誤僅與比較y_train(and y_val) 與logits/ softmax/label_estimations或您想要呼叫它們的任何內容有關。如上所述，通過顯式使用classes引數來設定。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/530662.html
上一篇：將numpy陣列的大小從(32,32,3)調整為(224,224,3)
下一篇：列印陣列串列的列