Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add attention-unet to the models #283

Closed
hvgazula opened this issue Mar 5, 2024 · 2 comments · Fixed by #295
Closed

Add attention-unet to the models #283

hvgazula opened this issue Mar 5, 2024 · 2 comments · Fixed by #295
Assignees
Labels
model Add new models to the API

Comments

@hvgazula
Copy link
Contributor

hvgazula commented Mar 5, 2024

@hvgazula hvgazula added the model Add new models to the API label Mar 5, 2024
@hvgazula hvgazula added this to Harsha Mar 10, 2024
@hvgazula hvgazula moved this to In Progress in Harsha Mar 10, 2024
@hvgazula
Copy link
Contributor Author

from torchinfo import summary
from models.networks.unet_grid_attention_3D import unet_grid_attention_3D
input_shape = (128, 128, 128)
batch_size = 1
image_channels = 3

summary(
    unet_grid_attention_3D(),
    input_size=(batch_size, image_channels, *input_shape),
    col_names=["input_size", "output_size", "num_params"],
    # depth=5,
)
  • The closes tensorflow/keras implementation of this paper is at https://github.com/robinvvinod/unet
  • however, it uses inception blocks. Probably, get this running first and them replace inception blocks with unet cnv blocks (need to look into this further)
  • print summary as follows
from network import network
import tensorflow  as tf
from tensorflow.keras.layers import Input

input_dimensions = (128, 128, 128, 1)
input_img = Input(input_dimensions)
model = network(input_img)
model.summary()

@hvgazula hvgazula self-assigned this Mar 10, 2024
hvgazula added a commit that referenced this issue Mar 10, 2024
@hvgazula
Copy link
Contributor Author

  • add tests and update __init__.py

@hvgazula hvgazula moved this from In Progress to Done in Harsha Mar 11, 2024
This was referenced Mar 12, 2024
Closed
Merged
@hvgazula hvgazula linked a pull request Mar 20, 2024 that will close this issue
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
model Add new models to the API
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant