/D-Unet_PyTorch

D-Unet in PyTorch

Primary LanguagePython

D-Unet: a dimension-fusion U shape network for chronic stroke lesion segmentation

D-Unet implemented in PyTorch

Usage

from DUnet import DUnet
import torch

BATCH_SIZE = 4
input_batch = torch.Tensor(BATCH_SIZE, 4, 192, 192)

model = DUnet(in_channels = 4)

output_batch = model(input_batch) # output_batch size : (BATCH_SIZE, 1, 192, 192)
  • According to the Reference paper input size must be (4, 192, 192) and output size must be (1, 192, 192)

Acknowledgment

  1. Special thanks to Seung hyun Hwang. He briefly reviewed the entire codes in this repository.

Reference

[1] Yongjin Zhou et al., D-UNet: a dimension-fusion U shape network for chronic stroke lesion segmentation ( arXiv:1908.05104 [eess.IV] ), 2019 Aug

[2] SZUHvern github source code implemented with keras (https://github.com/SZUHvern/D-UNet)