keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
998 stars 327 forks source link

Implement dropblock_2d regularization layer in keras 3 #2135

Closed divyashreepathihalli closed 8 months ago

divyashreepathihalli commented 10 months ago

keras-cv/keras_cv/layers/regularization/dropblock_2d.py

sami-bg commented 9 months ago

Do you mind if I give this a shot? It will be my first issue; I will be transliterating the layer's tf ops into the relevant keras ops. I will also write tests to make sure both implementations (keras 2 using TF and keras 3 using keras ops) match.

divyashreepathihalli commented 9 months ago

@sami-bg thank you for your interest. This layer would not be straightforward because it is subclassing an Keras 2 legacy internal baseRandomLayer. It could be a complicated issue for the first time.

sami-bg commented 9 months ago

Oh I see, I'll try for another issue then. Thanks for letting me know!

divyashreepathihalli commented 9 months ago

No worries! If you are interested we would love to have as many computer vision keras io examples converted to keras 3. This is a priority and could be a good one to start with. - https://github.com/keras-team/keras-io/tree/master/examples/vision

here is a list of them we are yet to convert

sami-bg commented 9 months ago

I'll give it a try! Thank you for the suggestion :)