語義細分的pytorch
該存儲庫包含一些用於語義細分的模型以及在Pytorch中實施的培訓和測試模型的管道
型號
- Vanilla FCN:FCN32,FCN16,FCN8,分別為VGG,Resnet和Densenet(用於語義分割的完全卷積網絡)
- U-NET(U-NET:生物醫學圖像分割的捲積網絡)
- segnet(Segnet:用於圖像分割的深卷積編碼器架構)
- PSPNET(金字塔場景解析網絡)
- GCN(大內核很重要)
- DUC,HDC(了解語義分割的捲積)
要求
- Pytorch 0.2.0
- Pytorch的張板。在這里安裝
- 其他一些庫(在運行代碼時找到您錯過的內容:p)
準備
- 轉到型號目錄,並在config.py中設置驗證模型的路徑
- 轉到數據集目錄並遵循讀取文件
托多
- DeepLab V3
- 煉油
- 更多數據集(例如ADE)