解決DHSNet ValueError: operands could not be broadcast together with shapes (224,224) (3, ) (224,224)

最近在復現DHSnet-pytorch模型,代碼調試過程中遇到如下錯誤: ValueError: operands could not be broadcast together with shapes (224,224) (3, ) (224,224) 通過錯誤描述可知該問題出現在 dataset.py 文件的第56行: 該函數的功能是將輸入圖片和ground truth 進行真值處理,以增強物
相關文章
相關標籤/搜索