UNet教學視頻

https://www.bilibili.com/video/BV11341127iK/?spm_id_from=333.337.search-card.all.click
Pytorch實現UNet代碼已經足夠了，看UNet的結構示意圖也足夠了

PASCAL VOC2012數據集簡介

https://blog.csdn.net/qq_37541097/article/details/115787033
剛好簡介了圖像分割任務和其它視覺任務的基本概念

調試代碼的問題記錄

• 我對原始代碼稍微修改，增加了大量注釋，歡迎關注收藏點贊github
  https://github.com/ningmengzhihe/UNet.git

• CUDA Out of Memory

試著調小batch_size

data_loader = DataLoader(MyDataset(data_path), batch_size=8, shuffle=True)

• 打印出來的圖片不一樣？

具體描述是這樣的：程序設置了每個epoch里邊每隔50個batch記錄當前batch的第一張（原始圖片、正確分割結果、當前算法分割結果）圖片對拼接，保存文件名稱是epoch_batch.png，那么是不是0_0.png和1_0.png圖片是同一張圖片呢？

如果shuffle設置成True，那么0_0.png和1_0.png不是同一張圖片，可能是shuffle設置成了隨機的原因，如下圖所示：

data_loader = DataLoader(MyDataset(data_path), batch_size=8, shuffle=True)

如果shuffle設置成False，那么0_0.png和1_0.png是同一張圖片了，可能是限制了每個batch中圖片的順序固定不變，如下圖所示

data_loader = DataLoader(MyDataset(data_path), batch_size=8, shuffle=False)

今天的分享就到這里了，感興趣的小伙伴快來點贊、收藏、轉發喲

總結

以上是生活随笔為你收集整理的用UNet做PASCAL VOC2012图像分割的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。