Skip to content

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

Notifications You must be signed in to change notification settings

remtav/EfficientUnet-PyTorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EfficientUnet-PyTorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

Useful notes

  1. Due to some rounding problem in PyTorch (not so sure), the input shape should be divisible by 32.
    e.g. 224x224 is a suitable size for input images, but 225x225 is not.

About

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%