Skip to content

Podidiving/unet_plus_attention

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Unet plus self attention


Implementation of vanilla unet & unet with self-attention

In self-attention implementation conv3x3 blocks are replaced by conv1x1+self-attention+conv1x1. See unet_plus_attention/model/model.py & unet_plus_attention/model/unet_blocks_builder.py for more details

About

Unet model with stand-alone self-attention blocks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published