keras-unet-collection icon indicating copy to clipboard operation
keras-unet-collection copied to clipboard

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.

Results 29 keras-unet-collection issues
Sort by recently updated
recently updated
newest added

If using the, for example, VGG19 backend, the [corresponding keras guide](https://keras.io/api/applications/vgg/) recommends applying the preprocessing from [here](https://www.tensorflow.org/api_docs/python/tf/keras/applications/vgg19/preprocess_input): > Note: each Keras Application expects a specific kind of input preprocessing. For...

documentation

Here is an example code for the dropout layer: ```python def GiveMeUnet(inputImage, numFilters = 16, droupouts = 0.1, doBatchNorm = True): # defining encoder Path c1 = Conv2dBlock(inputImage, numFilters *...

enhancement

Dear all, I hope you're going well. I would like to use the given exemple using Unet3p on Oxford Pet dataset with a dataset containing gray images as input (masks...

Love the project! I have tested it a while back but when I came back to the project yesterday I had a hard time remembering if it was hosted on...

I am trying figure out which dimension of the model output array on U2-Net and U-Net3+ is the final output. The models have multiple activation layers and the outputs from...

I need your help. I run the example file "Oxford_swin_Unet.ipynb" and then, I got an error "ModuleNotFoundError: No module named 'tensorflow.image'" how to fix it? ![image](https://user-images.githubusercontent.com/48307173/207470271-a0e280b0-e8d4-4b52-a2df-8ded827924ad.png)

add dropout support for basic CONV_Stack block.

Cannot assign value to variable ' block1_conv1/kernel:0': Shape mismatch.The variable shape (3, 3, 1, 64), and the assigned value shape (64, 3, 3, 3) are incompatible. File "D:\pneumothorax\Neel-Work_pnuemo\unet_plus_plus_lib.py", line 109,...

while running the resUnet_a_2d model i got this error -- File /opt/conda/lib/python3.10/site-packages/keras_unet_collection/_model_resunet_a_2d.py:177, in resunet_a_2d_base(input_tensor, filter_num, dilation_num, aspp_num_down, aspp_num_up, activation, batch_norm, pool, unpool, name) 173 X = ResUNET_a_block(X, f, kernel_size=3, dilation_num=dilation_[ind_],...