Skip to content

PyTorch implementation of the two U-Net-based architectures described in "Deep Learning for Segmentation using an Open Large-Scale Dataset in 2D Echocardiography"

License

Notifications You must be signed in to change notification settings

GKalliatakis/camus-segmentation-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CAMUS Segmentation: Semantic segmentation with PyTorch using an open large-scale dataset in 2D Echocardiography

Detailed architectures

The two U-Net implementations presented in the paper are summarised in tables I and II below. Both differ from the original U-Net proposed by Ronneberger et al. U-Net 1 is optimised for speed, while U-Net 2 is optimized for accuracy. U-Net 1 shows a more compact architecture, with the addition of one downsampling level. In the U-Net 2 design, the number of filters per convolutional layers increases and decreases linearly from an initial kernel size of 48, which makes for a wider net.

detailed_architectures

U-Net2

Usage

import torch
from camus_unet.camus_unet1 import CamusUnet1

if torch.cuda.is_available():
    device = torch.device('cuda')
else:
    device = torch.device('cpu')

model = CamusUnet1()  # initialise the U-NET 1 model

# move initialised model to chosen device
model = model.to(device)

# the usual training loop goes here...

References

License

Distributed under the MIT License. See LICENSE file for more information.

About

PyTorch implementation of the two U-Net-based architectures described in "Deep Learning for Segmentation using an Open Large-Scale Dataset in 2D Echocardiography"

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages