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

Remove bounding box utils and refactor retinanet #2039

Open
wants to merge 11 commits into
base: master
Choose a base branch
from
1 change: 0 additions & 1 deletion keras_hub/api/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
since your modifications would be overwritten.
"""

from keras_hub.api import bounding_box
from keras_hub.api import layers
from keras_hub.api import metrics
from keras_hub.api import models
Expand Down
23 changes: 0 additions & 23 deletions keras_hub/api/bounding_box/__init__.py

This file was deleted.

6 changes: 5 additions & 1 deletion keras_hub/api/layers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,16 @@
"""

from keras_hub.src.layers.modeling.alibi_bias import AlibiBias
from keras_hub.src.layers.modeling.anchor_generator import AnchorGenerator
from keras_hub.src.layers.modeling.box_matcher import BoxMatcher
from keras_hub.src.layers.modeling.cached_multi_head_attention import (
CachedMultiHeadAttention,
)
from keras_hub.src.layers.modeling.f_net_encoder import FNetEncoder
from keras_hub.src.layers.modeling.masked_lm_head import MaskedLMHead
from keras_hub.src.layers.modeling.non_max_supression import (
NonMaxSuppression as NonMaxSupression,
sineeli marked this conversation as resolved.
Show resolved Hide resolved
)
from keras_hub.src.layers.modeling.position_embedding import PositionEmbedding
from keras_hub.src.layers.modeling.reversible_embedding import (
ReversibleEmbedding,
Expand Down Expand Up @@ -55,7 +60,6 @@
from keras_hub.src.models.resnet.resnet_image_converter import (
ResNetImageConverter,
)
from keras_hub.src.models.retinanet.anchor_generator import AnchorGenerator
from keras_hub.src.models.retinanet.retinanet_image_converter import (
RetinaNetImageConverter,
)
Expand Down
2 changes: 0 additions & 2 deletions keras_hub/src/bounding_box/__init__.py

This file was deleted.

Loading
Loading