Add pretrained weights for CSPDarkNetTiny (#1251)

Author: ianstenbit

examples/training/classification/imagenet/training_history.json

@@ -20,6 +20,25 @@
             "validation_accuracy": "0.7744"
         }
     },
+    "cspdarknettiny": {
+        "v0": {
+            "accelerators": 8,
+            "args": {
+                "batch_size": "64",
+                "initial_learning_rate": "0.0125",
+                "use_ema": "True",
+                "weight_decay": "0.0001"
+            },
+            "contributor": "ianstenbit",
+            "epochs_trained": 136,
+            "script": {
+                "name": "basic_training.py",
+                "version": "212e67fc9acb65c699b609e4cdae54552d22e6b4"
+            },
+            "tensorboard_logs": "https://tensorboard.dev/experiment/yaJKJ934QO2i9ozLFrnfZw/",
+            "validation_accuracy": "0.6169"
+        }
+    },
     "darknet53": {
         "v0": {
             "accelerators": 8,

keras_cv/models/csp_darknet.py

@@ -272,7 +272,7 @@ def CSPDarkNetTiny(
         include_top=include_top,
         use_depthwise=use_depthwise,
         classes=classes,
-        weights=weights,
+        weights=parse_weights(weights, include_top, "cspdarknettiny"),
         input_shape=input_shape,
         input_tensor=input_tensor,
         pooling=pooling,
@@ -362,7 +362,7 @@ def CSPDarkNetL(
         include_top=include_top,
         use_depthwise=use_depthwise,
         classes=classes,
-        weights=parse_weights(weights, include_top, "cspdarknet"),
+        weights=parse_weights(weights, include_top, "cspdarknetl"),
         input_shape=input_shape,
         input_tensor=input_tensor,
         pooling=pooling,

keras_cv/models/weights.py

@@ -39,7 +39,11 @@ def parse_weights(weights, include_top, model_type):
 BASE_PATH = "https://storage.googleapis.com/keras-cv/models"
 
 ALIASES = {
-    "cspdarknet": {
+    "cspdarknetl": {
+        "imagenet": "imagenet/classification-v0",
+        "imagenet/classification": "imagenet/classification-v0",
+    },
+    "cspdarknettiny": {
         "imagenet": "imagenet/classification-v0",
         "imagenet/classification": "imagenet/classification-v0",
     },
@@ -113,10 +117,14 @@ def parse_weights(weights, include_top, model_type):
 }
 
 WEIGHTS_CONFIG = {
-    "cspdarknet": {
+    "cspdarknetl": {
         "imagenet/classification-v0": "8bdc3359222f0d26f77aa42c4e97d67a05a1431fe6c448ceeab9a9c5a34ff804",
         "imagenet/classification-v0-notop": "9303aabfadffbff8447171fce1e941f96d230d8f3cef30d3f05a9c85097f8f1e",
     },
+    "cspdarknettiny": {
+        "imagenet/classification-v0": "c17fe6d7b597f2eb25e42fbd97ec58fb1dad753ba18920cc27820953b7947704",
+        "imagenet/classification-v0-notop": "0007ae82c95be4d4aef06368a7c38e006381324d77e5df029b04890e18a8ad19",
+    },
     "darknet53": {
         "imagenet/classification-v0": "7bc5589f7f7f7ee3878e61ab9323a71682bfb617eb57f530ca8757c742f00c77",
         "imagenet/classification-v0-notop": "8dcce43163e4b4a63e74330ba1902e520211db72d895b0b090b6bfe103e7a8a5",