Register HGNetV2 presets (#2326)

Author: laxmareddyp

keras_hub/src/models/hgnetv2/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.hgnetv2.hgnetv2_backbone import HGNetV2Backbone
+from keras_hub.src.models.hgnetv2.hgnetv2_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, HGNetV2Backbone)

keras_hub/src/models/hgnetv2/hgnetv2_presets.py

@@ -9,7 +9,7 @@
             "params": 13599072,
             "path": "hgnetv2",
         },
-        "kaggle_handle": "",
+        "kaggle_handle": "kaggle://keras/hgnetv2/keras/hgnetv2_b4_ssld_stage2_ft_in1k/1",
     },
     "hgnetv2_b5_ssld_stage1_in22k_in1k": {
         "metadata": {
@@ -20,7 +20,7 @@
             "params": 33419680,
             "path": "hgnetv2",
         },
-        "kaggle_handle": "",
+        "kaggle_handle": "kaggle://keras/hgnetv2/keras/hgnetv2_b5_ssld_stage1_in22k_in1k/1",
     },
     "hgnetv2_b5_ssld_stage2_ft_in1k": {
         "metadata": {
@@ -31,7 +31,7 @@
             "params": 33419680,
             "path": "hgnetv2",
         },
-        "kaggle_handle": "",
+        "kaggle_handle": "kaggle://keras/hgnetv2/keras/hgnetv2_b5_ssld_stage2_ft_in1k/1",
     },
     "hgnetv2_b6_ssld_stage1_in22k_in1k": {
         "metadata": {
@@ -42,7 +42,7 @@
             "params": 69179888,
             "path": "hgnetv2",
         },
-        "kaggle_handle": "",
+        "kaggle_handle": "kaggle://keras/hgnetv2/keras/hgnetv2_b6_ssld_stage1_in22k_in1k/1",
     },
     "hgnetv2_b6_ssld_stage2_ft_in1k": {
         "metadata": {
@@ -53,6 +53,6 @@
             "params": 69179888,
             "path": "hgnetv2",
         },
-        "kaggle_handle": "",
+        "kaggle_handle": "kaggle://keras/hgnetv2/keras/hgnetv2_b6_ssld_stage2_ft_in1k/1",
     },
 }