セマンティックセグメンテーションのための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で優先モデルのパスを設定します
- DataSets Directoryに移動し、ReadMeに従います
トト
- DeepLab V3
- Refinenet
- その他のデータセット(ADEなど)