Skip to content

Commit abe1c79

Browse files
divyashreepathihalliushareng
authored andcommitted
update presets to point to the main Keras Kaggle page (keras-team#1921)
* update presets to point to the main keras page * update mit path
1 parent 611f418 commit abe1c79

File tree

7 files changed

+82
-26
lines changed

7 files changed

+82
-26
lines changed

keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,6 @@
1313
"path": "deeplabv3",
1414
"model_card": "https://arxiv.org/abs/1802.02611",
1515
},
16-
"kaggle_handle": "kaggle://kerashub/deeplabv3/keras/deeplabv3_plus_resnet50_pascalvoc",
16+
"kaggle_handle": "kaggle://keras/deeplabv3/keras/deeplabv3_plus_resnet50_pascalvoc/3",
1717
},
1818
}

keras_hub/src/models/densenet/densenet_presets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
"path": "densenet",
1313
"model_card": "https://arxiv.org/abs/1608.06993",
1414
},
15-
"kaggle_handle": "kaggle://kerashub/densenet/keras/densenet_121_imagenet",
15+
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/1",
1616
},
1717
"densenet_169_imagenet": {
1818
"metadata": {
@@ -25,7 +25,7 @@
2525
"path": "densenet",
2626
"model_card": "https://arxiv.org/abs/1608.06993",
2727
},
28-
"kaggle_handle": "kaggle://kerashub/densenet/keras/densenet_169_imagenet",
28+
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/1",
2929
},
3030
"densenet_201_imagenet": {
3131
"metadata": {
@@ -38,6 +38,6 @@
3838
"path": "densenet",
3939
"model_card": "https://arxiv.org/abs/1608.06993",
4040
},
41-
"kaggle_handle": "kaggle://kerashub/densenet/keras/densenet_201_imagenet",
41+
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/1",
4242
},
4343
}

keras_hub/src/models/mix_transformer/mix_transformer_presets.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
"official_name": "MiT",
2222
"path": "mit",
2323
},
24-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b0_ade20k_512",
24+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b0_ade20k_512/1",
2525
},
2626
"mit_b1_ade20k_512": {
2727
"metadata": {
@@ -32,7 +32,7 @@
3232
"official_name": "MiT",
3333
"path": "mit",
3434
},
35-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b1_ade20k_512",
35+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b1_ade20k_512/1",
3636
},
3737
"mit_b2_ade20k_512": {
3838
"metadata": {
@@ -43,7 +43,7 @@
4343
"official_name": "MiT",
4444
"path": "mit",
4545
},
46-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b2_ade20k_512",
46+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b2_ade20k_512/1",
4747
},
4848
"mit_b3_ade20k_512": {
4949
"metadata": {
@@ -54,7 +54,7 @@
5454
"official_name": "MiT",
5555
"path": "mit",
5656
},
57-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b3_ade20k_512",
57+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b3_ade20k_512/1",
5858
},
5959
"mit_b4_ade20k_512": {
6060
"metadata": {
@@ -65,7 +65,7 @@
6565
"official_name": "MiT",
6666
"path": "mit",
6767
},
68-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b4_ade20k_512",
68+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b4_ade20k_512/1",
6969
},
7070
"mit_b5_ade20k_640": {
7171
"metadata": {
@@ -76,7 +76,7 @@
7676
"official_name": "MiT",
7777
"path": "mit",
7878
},
79-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b5_ade20k_512",
79+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b5_ade20k_512/1",
8080
},
8181
"mit_b0_cityscapes_1024": {
8282
"metadata": {
@@ -87,7 +87,7 @@
8787
"official_name": "MiT",
8888
"path": "mit",
8989
},
90-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b0_cityscapes_1024",
90+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b0_cityscapes_1024/1",
9191
},
9292
"mit_b1_cityscapes_1024": {
9393
"metadata": {
@@ -98,7 +98,7 @@
9898
"official_name": "MiT",
9999
"path": "mit",
100100
},
101-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b1_cityscapes_1024",
101+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b1_cityscapes_1024/1",
102102
},
103103
"mit_b2_cityscapes_1024": {
104104
"metadata": {
@@ -109,7 +109,7 @@
109109
"official_name": "MiT",
110110
"path": "mit",
111111
},
112-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b2_cityscapes_1024",
112+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b2_cityscapes_1024/1",
113113
},
114114
"mit_b3_cityscapes_1024": {
115115
"metadata": {
@@ -120,7 +120,7 @@
120120
"official_name": "MiT",
121121
"path": "mit",
122122
},
123-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b3_cityscapes_1024",
123+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b3_cityscapes_1024/1",
124124
},
125125
"mit_b4_cityscapes_1024": {
126126
"metadata": {
@@ -131,7 +131,7 @@
131131
"official_name": "MiT",
132132
"path": "mit",
133133
},
134-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b4_cityscapes_1024",
134+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b4_cityscapes_1024/1",
135135
},
136136
"mit_b5_cityscapes_1024": {
137137
"metadata": {
@@ -142,7 +142,7 @@
142142
"official_name": "MiT",
143143
"path": "mit",
144144
},
145-
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b5_cityscapes_1024",
145+
"kaggle_handle": "kaggle://keras/mit/keras/mit_b5_cityscapes_1024/1",
146146
},
147147
}
148148

keras_hub/src/models/resnet/resnet_presets.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
"path": "resnet",
1313
"model_card": "https://arxiv.org/abs/2110.00476",
1414
},
15-
"kaggle_handle": "kaggle://kerashub/resnetv1/keras/resnet_18_imagenet/3",
15+
"kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_18_imagenet/1",
1616
},
1717
"resnet_50_imagenet": {
1818
"metadata": {
@@ -25,7 +25,7 @@
2525
"path": "resnet",
2626
"model_card": "https://arxiv.org/abs/2110.00476",
2727
},
28-
"kaggle_handle": "kaggle://kerashub/resnetv1/keras/resnet_50_imagenet/3",
28+
"kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_50_imagenet/1",
2929
},
3030
"resnet_101_imagenet": {
3131
"metadata": {
@@ -38,7 +38,7 @@
3838
"path": "resnet",
3939
"model_card": "https://arxiv.org/abs/2110.00476",
4040
},
41-
"kaggle_handle": "kaggle://kerashub/resnetv1/keras/resnet_101_imagenet/3",
41+
"kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_101_imagenet/1",
4242
},
4343
"resnet_152_imagenet": {
4444
"metadata": {
@@ -51,7 +51,7 @@
5151
"path": "resnet",
5252
"model_card": "https://arxiv.org/abs/2110.00476",
5353
},
54-
"kaggle_handle": "kaggle://kerashub/resnetv1/keras/resnet_152_imagenet/3",
54+
"kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_152_imagenet/1",
5555
},
5656
"resnet_v2_50_imagenet": {
5757
"metadata": {
@@ -64,7 +64,7 @@
6464
"path": "resnet",
6565
"model_card": "https://arxiv.org/abs/2110.00476",
6666
},
67-
"kaggle_handle": "kaggle://kerashub/resnetv2/keras/resnet_v2_50_imagenet/3",
67+
"kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_50_imagenet/1",
6868
},
6969
"resnet_v2_101_imagenet": {
7070
"metadata": {
@@ -77,6 +77,6 @@
7777
"path": "resnet",
7878
"model_card": "https://arxiv.org/abs/2110.00476",
7979
},
80-
"kaggle_handle": "kaggle://kerashub/resnetv2/keras/resnet_v2_101_imagenet/3",
80+
"kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_101_imagenet/1",
8181
},
8282
}

keras_hub/src/models/sam/sam_presets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
"path": "sam",
1010
"model_card": "https://arxiv.org/abs/2304.02643",
1111
},
12-
"kaggle_handle": "kaggle://kerashub/sam/keras/sam_base_sa1b/2",
12+
"kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/2",
1313
},
1414
"sam_large_sa1b": {
1515
"metadata": {
@@ -19,7 +19,7 @@
1919
"path": "sam",
2020
"model_card": "https://arxiv.org/abs/2304.02643",
2121
},
22-
"kaggle_handle": "kaggle://kerashub/sam/keras/sam_large_sa1b/2",
22+
"kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/3",
2323
},
2424
"sam_huge_sa1b": {
2525
"metadata": {
@@ -29,6 +29,6 @@
2929
"path": "sam",
3030
"model_card": "https://arxiv.org/abs/2304.02643",
3131
},
32-
"kaggle_handle": "kaggle://kerashub/sam/keras/sam_huge_sa1b/2",
32+
"kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/3",
3333
},
3434
}

keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,6 @@
1313
"path": "stablediffusion3",
1414
"model_card": "https://arxiv.org/abs/2110.00476",
1515
},
16-
"kaggle_handle": "kaggle://kerashub/stablediffusion3/keras/stable_diffusion_3_medium/4",
16+
"kaggle_handle": "kaggle://keras/stablediffusion3/keras/stable_diffusion_3_medium/1",
1717
}
1818
}
Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
"""vgg preset configurations."""
2+
3+
backbone_presets = {
4+
"vgg_11_imagenet": {
5+
"metadata": {
6+
"description": (
7+
"11-layer vgg model pre-trained on the ImageNet 1k dataset "
8+
"at a 224x224 resolution."
9+
),
10+
"params": 9220480,
11+
"official_name": "vgg",
12+
"path": "vgg",
13+
"model_card": "https://arxiv.org/abs/1409.1556",
14+
},
15+
"kaggle_handle": "kaggle://keras/vgg/keras/vgg_11_imagenet/1",
16+
},
17+
"vgg_13_imagenet": {
18+
"metadata": {
19+
"description": (
20+
"13-layer vgg model pre-trained on the ImageNet 1k dataset "
21+
"at a 224x224 resolution."
22+
),
23+
"params": 9404992,
24+
"official_name": "vgg",
25+
"path": "vgg",
26+
"model_card": "https://arxiv.org/abs/1409.1556",
27+
},
28+
"kaggle_handle": "kaggle://keras/vgg/keras/vgg_13_imagenet/1",
29+
},
30+
"vgg_16_imagenet": {
31+
"metadata": {
32+
"description": (
33+
"16-layer vgg model pre-trained on the ImageNet 1k dataset "
34+
"at a 224x224 resolution."
35+
),
36+
"params": 14714688,
37+
"official_name": "vgg",
38+
"path": "vgg",
39+
"model_card": "https://arxiv.org/abs/1409.1556",
40+
},
41+
"kaggle_handle": "kaggle://keras/vgg/keras/vgg_16_imagenet/1",
42+
},
43+
"vgg_19_imagenet": {
44+
"metadata": {
45+
"description": (
46+
"19-layer vgg model pre-trained on the ImageNet 1k dataset "
47+
"at a 224x224 resolution."
48+
),
49+
"params": 20024384,
50+
"official_name": "vgg",
51+
"path": "vgg",
52+
"model_card": "https://arxiv.org/abs/1409.1556",
53+
},
54+
"kaggle_handle": "kaggle://keras/vgg/keras/vgg_19_imagenet/1",
55+
},
56+
}

0 commit comments

Comments
 (0)