Skip to content

Latest commit

 

History

History
39 lines (18 loc) · 920 Bytes

README.md

File metadata and controls

39 lines (18 loc) · 920 Bytes
Raw

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)