Skip to content

Commit eae2c0d

Browse files
authored
Merge branch 'master' into port_mobilenet
2 parents fe34e36 + abf7432 commit eae2c0d

37 files changed

+323
-2455
lines changed

keras_hub/api/__init__.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
since your modifications would be overwritten.
55
"""
66

7-
from keras_hub.api import bounding_box
87
from keras_hub.api import layers
98
from keras_hub.api import metrics
109
from keras_hub.api import models

keras_hub/api/bounding_box/__init__.py

-23
This file was deleted.

keras_hub/api/layers/__init__.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,14 @@
55
"""
66

77
from keras_hub.src.layers.modeling.alibi_bias import AlibiBias
8+
from keras_hub.src.layers.modeling.anchor_generator import AnchorGenerator
9+
from keras_hub.src.layers.modeling.box_matcher import BoxMatcher
810
from keras_hub.src.layers.modeling.cached_multi_head_attention import (
911
CachedMultiHeadAttention,
1012
)
1113
from keras_hub.src.layers.modeling.f_net_encoder import FNetEncoder
1214
from keras_hub.src.layers.modeling.masked_lm_head import MaskedLMHead
15+
from keras_hub.src.layers.modeling.non_max_supression import NonMaxSuppression
1316
from keras_hub.src.layers.modeling.position_embedding import PositionEmbedding
1417
from keras_hub.src.layers.modeling.reversible_embedding import (
1518
ReversibleEmbedding,
@@ -58,7 +61,6 @@
5861
from keras_hub.src.models.resnet.resnet_image_converter import (
5962
ResNetImageConverter,
6063
)
61-
from keras_hub.src.models.retinanet.anchor_generator import AnchorGenerator
6264
from keras_hub.src.models.retinanet.retinanet_image_converter import (
6365
RetinaNetImageConverter,
6466
)

keras_hub/api/models/__init__.py

+9-4
Original file line numberDiff line numberDiff line change
@@ -193,10 +193,6 @@
193193
from keras_hub.src.models.image_classifier_preprocessor import (
194194
ImageClassifierPreprocessor,
195195
)
196-
from keras_hub.src.models.image_object_detector import ImageObjectDetector
197-
from keras_hub.src.models.image_object_detector_preprocessor import (
198-
ImageObjectDetectorPreprocessor,
199-
)
200196
from keras_hub.src.models.image_segmenter import ImageSegmenter
201197
from keras_hub.src.models.image_segmenter_preprocessor import (
202198
ImageSegmenterPreprocessor,
@@ -234,6 +230,15 @@
234230
)
235231
from keras_hub.src.models.mobilenet.mobilenet_image_classifier_preprocessor import (
236232
MobileNetImageClassifierPreprocessor,
233+
from keras_hub.src.models.object_detector import ObjectDetector
234+
from keras_hub.src.models.object_detector import (
235+
ObjectDetector as ImageObjectDetector,
236+
)
237+
from keras_hub.src.models.object_detector_preprocessor import (
238+
ObjectDetectorPreprocessor,
239+
)
240+
from keras_hub.src.models.object_detector_preprocessor import (
241+
ObjectDetectorPreprocessor as ImageObjectDetectorPreprocessor,
237242
)
238243
from keras_hub.src.models.opt.opt_backbone import OPTBackbone
239244
from keras_hub.src.models.opt.opt_causal_lm import OPTCausalLM

keras_hub/src/bounding_box/__init__.py

-2
This file was deleted.

0 commit comments

Comments
 (0)