Skip to content

Commit b148fe6

Browse files
divyashreepathihallidivyashreepathihalli
authored
Add KerasHub model symbols (#1951)
* add KerasHub model symbols * correct dense spelling * update sam layers * fix error * fix tests * correct typo
1 parent db270c7 commit b148fe6

File tree

1 file changed

+186
-1
lines changed

1 file changed

+186
-1
lines changed

‎scripts/hub_api_master.py‎

Lines changed: 186 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -971,6 +971,191 @@
971971
},
972972
],
973973
},
974+
{
975+
"path": "densnet/",
976+
"title": "DenseNet",
977+
"toc": True,
978+
"children": [
979+
{
980+
"path": "densenet_image_converter",
981+
"title": "DenseNetImageConverter",
982+
"generate": [
983+
"keras_hub.layers.DenseNetImageConverter",
984+
"keras_hub.layers.DenseNetImageConverter.from_preset",
985+
],
986+
},
987+
{
988+
"path": "densenet_backbone",
989+
"title": "DensNetBackbone model",
990+
"generate": [
991+
"keras_hub.models.DenseNetBackbone",
992+
"keras_hub.models.DenseNetBackbone.from_preset",
993+
"keras_hub.models.DenseNetBackbone.token_embedding",
994+
],
995+
},
996+
{
997+
"path": "densenet_image_classifier",
998+
"title": "DenseNetImageClassifier model",
999+
"generate": [
1000+
"keras_hub.models.DenseNetImageClassifier",
1001+
"keras_hub.models.DenseNetImageClassifier.from_preset",
1002+
"keras_hub.models.DenseNetImageClassifier.backbone",
1003+
"keras_hub.models.DenseNetImageClassifier.preprocessor",
1004+
],
1005+
},
1006+
{
1007+
"path": "densenet_image_classifier_preprocessor",
1008+
"title": "DenseNetImageClassifierPreprocessor layer",
1009+
"generate": [
1010+
"keras_hub.models.DenseNetImageClassifierPreprocessor",
1011+
"keras_hub.models.DenseNetImageClassifierPreprocessor.from_preset",
1012+
"keras_hub.models.DenseNetImageClassifierPreprocessor.image_converter",
1013+
],
1014+
},
1015+
],
1016+
},
1017+
{
1018+
"path": "sam/",
1019+
"title": "Segment Anything Model",
1020+
"toc": True,
1021+
"children": [
1022+
{
1023+
"path": "sam_image_converter",
1024+
"title": "SAMImageConverter",
1025+
"generate": [
1026+
"keras_hub.layers.SAMImageConverter",
1027+
"keras_hub.layers.SAMImageConverter.from_preset",
1028+
],
1029+
},
1030+
{
1031+
"path": "sam_backbone",
1032+
"title": "SAMBackbone model",
1033+
"generate": [
1034+
"keras_hub.models.SAMBackbone",
1035+
"keras_hub.models.SAMBackbone.from_preset",
1036+
"keras_hub.models.SAMBackbone.token_embedding",
1037+
],
1038+
},
1039+
{
1040+
"path": "sam_image_segmenter",
1041+
"title": "SAMImageSegmenter model",
1042+
"generate": [
1043+
"keras_hub.models.SAMImageSegmenter",
1044+
"keras_hub.models.SAMImageSegmenter.from_preset",
1045+
"keras_hub.models.SAMImageSegmenter.backbone",
1046+
"keras_hub.models.SAMImageSegmenter.preprocessor",
1047+
],
1048+
},
1049+
{
1050+
"path": "sam_image_segmenter_preprocessor",
1051+
"title": "SAMImageSegmenterPreprocessor layer",
1052+
"generate": [
1053+
"keras_hub.models.SAMImageSegmenterPreprocessor",
1054+
"keras_hub.models.SAMImageSegmenterPreprocessor.from_preset",
1055+
"keras_hub.models.SAMImageSegmenterPreprocessor.image_converter",
1056+
],
1057+
},
1058+
{
1059+
"path": "sam_prompt_encoder",
1060+
"title": "SAMPromptEncoder layer",
1061+
"generate": [
1062+
"keras_hub.layers.SAMPromptEncoder",
1063+
],
1064+
},
1065+
{
1066+
"path": "sam_mask_decoder",
1067+
"title": "SAMMaskDecoder layer",
1068+
"generate": [
1069+
"keras_hub.layers.SAMMaskDecoder",
1070+
],
1071+
},
1072+
],
1073+
},
1074+
{
1075+
"path": "stable_diffusion_3/",
1076+
"title": "Stable Diffusion 3",
1077+
"toc": True,
1078+
"children": [
1079+
{
1080+
"path": "sam_image_converter",
1081+
"title": "SAMImageConverter",
1082+
"generate": [
1083+
"keras_hub.layers.SAMImageConverter",
1084+
"keras_hub.layers.SAMImageConverter.from_preset",
1085+
],
1086+
},
1087+
{
1088+
"path": "stable_diffusion_3_backbone",
1089+
"title": "StableDiffusion3Backbone model",
1090+
"generate": [
1091+
"keras_hub.models.StableDiffusion3Backbone",
1092+
"keras_hub.models.StableDiffusion3Backbone.from_preset",
1093+
"keras_hub.models.StableDiffusion3Backbone.token_embedding",
1094+
],
1095+
},
1096+
{
1097+
"path": "stable_diffusion_3_text_to_image",
1098+
"title": "StableDiffusion3TextToImage model",
1099+
"generate": [
1100+
"keras_hub.models.StableDiffusion3TextToImage",
1101+
"keras_hub.models.StableDiffusion3TextToImage.from_preset",
1102+
"keras_hub.models.StableDiffusion3TextToImage.backbone",
1103+
"keras_hub.models.StableDiffusion3TextToImage.preprocessor",
1104+
],
1105+
},
1106+
{
1107+
"path": "stable_diffusion_3_text_to_image_preprocessor",
1108+
"title": "StableDiffusion3TextToImagePreprocessor layer",
1109+
"generate": [
1110+
"keras_hub.models.StableDiffusion3TextToImagePreprocessor",
1111+
"keras_hub.models.StableDiffusion3TextToImagePreprocessor.from_preset",
1112+
],
1113+
},
1114+
],
1115+
},
1116+
{
1117+
"path": "deeplab_v3/",
1118+
"title": "DeepLabV3 and DeepLabV3Plus",
1119+
"toc": True,
1120+
"children": [
1121+
{
1122+
"path": "deeplab_v3_image_converter",
1123+
"title": "DeepLabV3ImageConverter",
1124+
"generate": [
1125+
"keras_hub.layers.DeepLabV3ImageConverter",
1126+
"keras_hub.layers.DeepLabV3ImageConverter.from_preset",
1127+
],
1128+
},
1129+
{
1130+
"path": "deeplab_v3_backbone",
1131+
"title": "DeepLabV3Backbone model",
1132+
"generate": [
1133+
"keras_hub.models.DeepLabV3Backbone",
1134+
"keras_hub.models.DeepLabV3Backbone.from_preset",
1135+
"keras_hub.models.DeepLabV3Backbone.token_embedding",
1136+
],
1137+
},
1138+
{
1139+
"path": "deeplab_v3_image_segmenter",
1140+
"title": "DeepLabV3ImageSegmenter model",
1141+
"generate": [
1142+
"keras_hub.models.DeepLabV3ImageSegmenter",
1143+
"keras_hub.models.DeepLabV3ImageSegmenter.from_preset",
1144+
"keras_hub.models.DeepLabV3ImageSegmenter.backbone",
1145+
"keras_hub.models.DeepLabV3ImageSegmenter.preprocessor",
1146+
],
1147+
},
1148+
{
1149+
"path": "deeplab_v3_image_segmenter_preprocessor",
1150+
"title": "DeepLabV3ImageSegmenterPreprocessor layer",
1151+
"generate": [
1152+
"keras_hub.models.DeepLabV3ImageSegmenterPreprocessor",
1153+
"keras_hub.models.DeepLabV3ImageSegmenterPreprocessor.from_preset",
1154+
"keras_hub.models.DeepLabV3ImageSegmenterPreprocessor.image_converter",
1155+
],
1156+
},
1157+
],
1158+
},
9741159
{
9751160
"path": "resnet/",
9761161
"title": "ResNet",
@@ -1009,7 +1194,7 @@
10091194
"generate": [
10101195
"keras_hub.models.ResNetImageClassifierPreprocessor",
10111196
"keras_hub.models.ResNetImageClassifierPreprocessor.from_preset",
1012-
"keras_hub.models.ResNetImageClassifierPreprocessor.tokenizer",
1197+
"keras_hub.models.ResNetImageClassifierPreprocessor.image_converter",
10131198
],
10141199
},
10151200
],

0 commit comments

Comments
 (0)