Skip to content

Commit 0cbaf0f

Browse files
Cspnet architecture. (#2091)
* cspnet architecture to support multiple models * add timm conversion script * add test case * fix format * fix testcase * add preset details. * add timm conversion test * register preprocessor * register presets * nit * fix label typo
1 parent 5456fa9 commit 0cbaf0f

19 files changed

+1781
-533
lines changed

keras_hub/api/layers/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,9 @@
4242
BASNetImageConverter,
4343
)
4444
from keras_hub.src.models.clip.clip_image_converter import CLIPImageConverter
45+
from keras_hub.src.models.cspnet.cspnet_image_converter import (
46+
CSPNetImageConverter,
47+
)
4548
from keras_hub.src.models.deeplab_v3.deeplab_v3_image_converter import (
4649
DeepLabV3ImageConverter,
4750
)

keras_hub/api/models/__init__.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,11 +61,12 @@
6161
from keras_hub.src.models.clip.clip_text_encoder import CLIPTextEncoder
6262
from keras_hub.src.models.clip.clip_tokenizer import CLIPTokenizer
6363
from keras_hub.src.models.clip.clip_vision_encoder import CLIPVisionEncoder
64-
from keras_hub.src.models.csp_darknet.csp_darknet_backbone import (
65-
CSPDarkNetBackbone,
64+
from keras_hub.src.models.cspnet.cspnet_backbone import CSPNetBackbone
65+
from keras_hub.src.models.cspnet.cspnet_image_classifier import (
66+
CSPNetImageClassifier,
6667
)
67-
from keras_hub.src.models.csp_darknet.csp_darknet_image_classifier import (
68-
CSPDarkNetImageClassifier,
68+
from keras_hub.src.models.cspnet.cspnet_image_classifier_preprocessor import (
69+
CSPNetImageClassifierPreprocessor,
6970
)
7071
from keras_hub.src.models.deberta_v3.deberta_v3_backbone import (
7172
DebertaV3Backbone,

keras_hub/src/models/csp_darknet/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)