原创 Pytorch入門——用UNet網絡做圖像分割

最近看的paper裏的pytorch代碼太複雜,我之前也沒接觸過pytorch,遂決定先自己實現一個基礎的裸代碼,這樣走一遍,對跑網絡的基本流程和一些常用的基礎函數的印象會更深刻。 本文的代碼和數據主要來自https://blog.csd