Segformer. AI model that understands what's in images and labels each part.

github.com/lucidrains/segformer-pytorch
by @lucidrains

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

August 2021
  • Release —0.0.6
  • fix layernorm
June 2021
  • Release —0.0.5
  • fix initial image to patches and remove patch_size hyperparameter
  • Release —0.0.4
  • use depth-wise conv
  • Release —0.0.3
  • use conv2d kernel and stride for reduction ratio
  • Release —0.0.2
  • make sure it works for any image size
  • Release —0.0.1
  • remove wip
  • Create python-publish.yml
  • complete segformer
  • add some scaffold
  • readme
  • Initial commit