The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
MIT License
660
stars
185
forks
source link
ValueError: Shapes (739, 128, 128, 3) and (128, 128, 3) are incompatible #52
I got below error when tried to use "transunet_2d"
model = models.transunet_2d((128, 128, 3), filter_num=[64, 128, 256, 512], n_labels=3, stack_num_down=2, stack_num_up=2, embed_dim=768, num_mlp=3072, num_heads=12, num_transformer=12, activation='ReLU', mlp_activation='GELU', output_activation='Softmax', batch_norm=True, pool=True, unpool='bilinear', name='transunet')
See below error:
24/24 [==============================] - 1975s 82s/step
ValueError Traceback (most recent call last)