ConvNeXtV2-Unet icon indicating copy to clipboard operation
ConvNeXtV2-Unet copied to clipboard

Unet-implementation based on ConvNeXtV2

ConvNeXtV2-Unet

pytorch, ConvNeXtV2,Unet

Adapted from https://github.com/facebookresearch/ConvNeXt-V2

Usage

from Unet import ConvNeXtV2
model = ConvNeXtV2()

x = torch.randn(1, 3, 256, 256)
y = model(x)