`Equalization` channel dimension is not calculated during trace time

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras

Other

1.01k stars 330 forks source link

`Equalization` channel dimension is not calculated during trace time #431

Open quantumalaviya opened 2 years ago

quantumalaviya commented 2 years ago

There are a few issues I faced while trying to use the Equalization layer with AugMix in #407

The channel dimension is not calculated during trace time (I believe due to the vectorized_map from #201). This makes it incompatible with other layers in a chain. Moreover, this is the only layer that gets traced every time it's called. It's not ideal and repeated occurrences of this layer are very inefficient.

LukeWood commented 2 years ago

Definitely, Equalization should be updated. Thanks for doing some digging here.

I believe these changes were introduced by my PR that adds support for dynamic channel sizes; but this behavior is very undesirable.

github-actions[bot] commented 10 months ago

This issue is stale because it has been open for 180 days with no activity. It will be closed if no further activity occurs. Thank you.