nikhilroxtomar / Semantic-Segmentation-Architecture

A repository contains the code for various semantic segmentation in TensorFlow and PyTorch framework.
Apache License 2.0
152 stars 66 forks source link

Update attention-unet.py #6

Closed chathurawimalasiri closed 1 month ago

chathurawimalasiri commented 3 months ago

These changes are made to handle variations in the input image size, ensuring that the tensors are compatible for operations like addition and concatenation. They dynamically adjust the dimensions of the tensors involved, maintaining the integrity of the model's computations without requiring fixed input dimensions. This way, the model can process images of varying sizes.