Rare find

X-UNet. A neural network tool for image and video analysis tasks.

github.com/lucidrains/x-unet
by @lucidrains

Vaya's read on this project

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

Updates

May 2024
  • Release —0.4.0
  • remove group norms https://arxiv.org/abs/2312.02696
July 2023
  • Release —0.3.1
  • able to customize attention heads and dim head differently across stages
January 2023
  • Release —0.3.0
  • switch 3x3 convs within convnext to be semi depthwise
  • Release —0.2.1
  • make it work with rectangular images
  • Merge pull request #6 from karolmajek/main
  • Release —0.2.0
  • update convnext with global response normalization from updated paper
  • add channels param to get rid of confusion
October 2022
  • switch back to regular attention
  • Release —0.1.0
September 2022
  • Release —0.0.22
  • make sure nested unet can accommodate feature dimensions other than p…
August 2022
  • Release —0.0.21
  • upgrade to best type of downsample
  • Release —0.0.20
  • change the nested RSU from unet squared to using conv3d, in preparati…
  • cleanup
  • cite
  • Release —0.0.19
  • add ability to finely specify number of self attention blocks per stage
  • add support for 3d
  • make sure cosine sim scale works for fp16, also make sure attention i…
  • apply weight standardization to the nested RSU from unet squared paper
  • add weight standardization
  • add diagram
  • add a scale to skip connection