Implementasi berbagai jenis model UNET untuk segmentasi gambar
Unet - u -net: jaringan konvolusional untuk segmentasi gambar biomedis https://arxiv.org/abs/1505.04597
RCNN-UNET -Jaringan saraf konvolusional residu berulang berdasarkan U-Net (R2U-net) untuk segmentasi gambar medis https://arxiv.org/abs/1802.06955
Perhatian UNET - Perhatian U -Net: Belajar di mana mencari pankreas https://arxiv.org/abs/1804.03999
RCNN-ATENSI UNET -PERHATIAN R2U-NET: Just Integration dari dua karya canggih baru-baru ini (R2U-net + Attention U-Net)
Dengan visualisasi lapisan
Kloning repo:
git clone https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets.git python>=3.6
torch>=0.4.0
torchvision
torchsummary
tensorboardx
natsort
numpy
pillow
scipy
scikit-image
sklearn
Instal semua perpustakaan dependen:
pip install -r requirements.txtTambahkan semua folder Anda ke baris ini 106-113
t_data = '' # Input data
l_data = '' #Input Label
test_image = '' #Image to be predicted while training
test_label = '' #Label of the prediction Image
test_folderP = '' #Test folder Image
test_folderL = '' #Test folder Label for calculating the Dice score
Tidak ada 
Rcnn unet 
Perhatian tidak ada 
Perhatian-rcnn unet 
Bersarang Unet

Untuk merencanakan kerugian, Visdom akan diperlukan. Kode sudah ditulis, hanya membatalkan komentar bagian yang diperlukan. Aliran gradien juga dapat digunakan. Diambil dari (https://discuss.pytorch.org/t/check-gradient-flow-in-network/15063/10)
Folder model dibuat dan semua data disimpan di dalamnya. Lapisan terakhir akan disimpan di folder model. Jika diperlukan lapisan tertentu, sebutkan di baris 361.
VISULASI LAYER

Visulisisasi Filter

Tensorboardx masih harus mengubah beberapa parameter untuk mendapatkan visualisasi. Telah mengacaukan ini mencoba membuat Pytorch 1.1.0 bekerja dengan Tensorboard secara langsung (dan kemudian mengetahui saat ini itu tidak mendukung apa pun selain dari grafik linier) 
Input Visulisisasi Gambar untuk Memeriksa
a) Gambar asli

b) Gambar Centercrop

Skor DICE untuk Segmentasi Hippocampus Dataset Adni-Loni

Jika Anda menemukan itu berguna untuk pekerjaan Anda.
@article{DBLP:journals/corr/abs-1906-07160,
author = {Malav Bateriwala and
Pierrick Bourgeat},
title = {Enforcing temporal consistency in Deep Learning segmentation of brain
{MR} images},
journal = {CoRR},
volume = {abs/1906.07160},
year = {2019},
url = {http://arxiv.org/abs/1906.07160},
archivePrefix = {arXiv},
eprint = {1906.07160},
timestamp = {Mon, 24 Jun 2019 17:28:45 +0200},
biburl = {https://dblp.org/rec/bib/journals/corr/abs-1906-07160},
bibsource = {dblp computer science bibliography, https://dblp.org}
}
In progress