Skip to content

Commit

Permalink
Update Models API
Browse files Browse the repository at this point in the history
  • Loading branch information
@sachinprasadhs
sachinprasadhs authored Mar 15, 2024
1 parent 54212d9 commit ede83b8
Showing 1 changed file with 193 additions and 41 deletions.
234 changes: 193 additions & 41 deletions scripts/cv_api_master.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -306,6 +306,108 @@
"title": "Backbones",
"toc": True,
"children": [
{
"path": "csp_darknet",
"title": "CSPDarkNet backbones",
"generate": [
"keras_cv.models.CSPDarkNetBackbone",
"keras_cv.models.CSPDarkNetBackbone.from_preset",
"keras_cv.models.CSPDarkNetTinyBackbone",
"keras_cv.models.CSPDarkNetSBackbone",
"keras_cv.models.CSPDarkNetMBackbone",
"keras_cv.models.CSPDarkNetLBackbone",
"keras_cv.models.CSPDarkNetXLBackbone",
],
},
{
"path": "densenet",
"title": "DenseNet backbones",
"generate": [
"keras_cv.models.DenseNetBackbone",
"keras_cv.models.DenseNetBackbone.from_preset",
"keras_cv.models.DenseNet121Backbone",
"keras_cv.models.DenseNet169Backbone",
"keras_cv.models.DenseNet201Backbone",
],
},
{
"path": "efficientnetv1",
"title": "EfficientNetV1 models",
"generate": [
"keras_cv.models.EfficientNetV1Backbone",
"keras_cv.models.EfficientNetV1Backbone.from_preset",
"keras_cv.models.EfficientNetV1B0Backbone",
"keras_cv.models.EfficientNetV1B1Backbone",
"keras_cv.models.EfficientNetV1B2Backbone",
"keras_cv.models.EfficientNetV1B3Backbone",
"keras_cv.models.EfficientNetV1B4Backbone",
"keras_cv.models.EfficientNetV1B5Backbone",
"keras_cv.models.EfficientNetV1B6Backbone",
"keras_cv.models.EfficientNetV1B7Backbone",
],
},
{
"path": "efficientnetv2",
"title": "EfficientNetV2 models",
"generate": [
"keras_cv.models.EfficientNetV2Backbone",
"keras_cv.models.EfficientNetV2Backbone.from_preset",
"keras_cv.models.EfficientNetV2B0Backbone",
"keras_cv.models.EfficientNetV2B1Backbone",
"keras_cv.models.EfficientNetV2B2Backbone",
"keras_cv.models.EfficientNetV2B3Backbone",
"keras_cv.models.EfficientNetV2SBackbone",
"keras_cv.models.EfficientNetV2MBackbone",
"keras_cv.models.EfficientNetV2LBackbone",
],
},
{
"path": "efficientnet_lite",
"title": "EfficientNet Lite backbones",
"generate": [
"keras_cv.models.EfficientNetLiteBackbone",
"keras_cv.models.EfficientNetLiteBackbone.from_preset",
"keras_cv.models.EfficientNetLiteB0Backbone",
"keras_cv.models.EfficientNetLiteB1Backbone",
"keras_cv.models.EfficientNetLiteB2Backbone",
"keras_cv.models.EfficientNetLiteB3Backbone",
"keras_cv.models.EfficientNetLiteB4Backbone",
],
},
{
"path": "mix_transformer",
"title": "MixTransformer backbones",
"generate": [
"keras_cv.models.MiTBackbone",
"keras_cv.models.MiTBackbone.from_preset",
"keras_cv.models.MiTB0Backbone",
"keras_cv.models.MiTB1Backbone",
"keras_cv.models.MiTB2Backbone",
"keras_cv.models.MiTB3Backbone",
"keras_cv.models.MiTB4Backbone",
"keras_cv.models.MiTB5Backbone",
],
},
{
"path": "mobilenetv3",
"title": "MobileNetV3 backbones",
"generate": [
"keras_cv.models.MobileNetV3Backbone",
"keras_cv.models.MobileNetV3Backbone.from_preset",
"keras_cv.models.MobileNetV3SmallBackbone",
"keras_cv.models.MobileNetV3LargeBackbone",
],
},
{
"path": "mobilenetv3",
"title": "MobileNetV3 backbones",
"generate": [
"keras_cv.models.MobileNetV3Backbone",
"keras_cv.models.MobileNetV3Backbone.from_preset",
"keras_cv.models.MobileNetV3SmallBackbone",
"keras_cv.models.MobileNetV3LargeBackbone",
],
},
{
"path": "resnet_v1",
"title": "ResNetV1 backbones",
Expand All @@ -332,50 +434,31 @@
"keras_cv.models.ResNet152V2Backbone",
],
},
# {
# "path": "csp_darknet",
# "title": "CSPDarkNet backbones",
# "generate": [
# "keras_cv.models.CSPDarkNetBackbone",
# "keras_cv.models.CSPDarkNetBackbone.from_preset",
# "keras_cv.models.CSPDarkNetTinyBackbone",
# "keras_cv.models.CSPDarkNetSBackbone",
# "keras_cv.models.CSPDarkNetMBackbone",
# "keras_cv.models.CSPDarkNetLBackbone",
# "keras_cv.models.CSPDarkNetXLBackbone",
# ],
# },
{
"path": "yolo_v8",
"title": "YOLOV8 backbones",
"path": "vgg16",
"title": "VGG16 backbones",
"generate": [
"keras_cv.models.YOLOV8Backbone",
"keras_cv.models.YOLOV8Backbone.from_preset",
"keras_cv.models.VGG16Backbone",
"keras_cv.models.VGG16Backbone.from_preset",
],
},
{
"path": "mobilenetv3",
"title": "MobileNetV3 backbones",
"path": "vitdet",
"title": "ViTDet backbones",
"generate": [
"keras_cv.models.MobileNetV3Backbone",
"keras_cv.models.MobileNetV3Backbone.from_preset",
"keras_cv.models.MobileNetV3SmallBackbone",
"keras_cv.models.MobileNetV3LargeBackbone",
"keras_cv.models.ViTDetBackbone",
"keras_cv.models.ViTDetBackbone.from_preset",
"keras_cv.models.ViTDetBBackbone",
"keras_cv.models.ViTDetLBackbone",
"keras_cv.models.ViTDetHBackbone",
],
},
{
"path": "efficientnetv2",
"title": "EfficientNetV2 models",
"path": "yolo_v8",
"title": "YOLOV8 backbones",
"generate": [
"keras_cv.models.EfficientNetV2Backbone",
"keras_cv.models.EfficientNetV2Backbone.from_preset",
"keras_cv.models.EfficientNetV2B0Backbone",
"keras_cv.models.EfficientNetV2B1Backbone",
"keras_cv.models.EfficientNetV2B2Backbone",
"keras_cv.models.EfficientNetV2B3Backbone",
"keras_cv.models.EfficientNetV2SBackbone",
"keras_cv.models.EfficientNetV2MBackbone",
"keras_cv.models.EfficientNetV2LBackbone",
"keras_cv.models.YOLOV8Backbone",
"keras_cv.models.YOLOV8Backbone.from_preset",
],
},
],
Expand All @@ -387,18 +470,63 @@
"toc": True,
"children": [
{
"path": "stable_diffusion",
"title": "StableDiffusion image-generation model",
"path": "basnet_segmentation",
"title": "BASNet Segmentation",
"generate": [
"keras_cv.models.StableDiffusion",
"keras_cv.models.BASNet",
"keras_cv.models.BASNet.from_preset",
],
},
{
"path": "retinanet",
"title": "The RetinaNet model",
"path": "deeplabv3_segmentation",
"title": "DeepLabV3Plus Segmentation",
"generate": [
"keras_cv.models.RetinaNet",
"keras_cv.models.RetinaNet.from_preset",
"keras_cv.models.DeepLabV3Plus",
"keras_cv.models.DeepLabV3Plus.from_preset",
],
},
{
"path": "segformer_segmentation",
"title": "SegFormer Segmentation",
"generate": [
"keras_cv.models.SegFormer",
"keras_cv.models.SegFormer.from_preset",
"keras_cv.models.SegFormerB0",
"keras_cv.models.SegFormerB1",
"keras_cv.models.SegFormerB2",
"keras_cv.models.SegFormerB3",
"keras_cv.models.SegFormerB4",
"keras_cv.models.SegFormerB5",
],
},
{
"path": "segment_anything",
"title": "Segment Anything",
"generate": [
"keras_cv.models.SegmentAnythingModel",
"keras_cv.models.SegmentAnythingModel.from_preset",
"keras_cv.models.SAMMaskDecoder",
"keras_cv.models.SAMPromptEncoder",
"keras_cv.models.TwoWayTransformer",
"keras_cv.layers.MultiHeadAttentionWithDownsampling",
"keras_cv.layers.TwoWayMultiHeadAttention",
"keras_cv.layers.RandomFrequencyPositionalEmbeddings",
],
},
{
"path": "feature_extractor",
"title": "CLIP Feature extractor",
"generate": [
"keras_cv.models.CLIP",
"keras_cv.models.CLIP.from_preset",
"keras_cv.models.feature_extractor.CLIPAttention",
"keras_cv.models.feature_extractor.CLIPEncoder",
"keras_cv.models.feature_extractor.CLIPImageEncoder",
"keras_cv.models.feature_extractor.CLIPPatchingAndEmbedding",
"keras_cv.models.feature_extractor.CLIPProcessor",
"keras_cv.models.feature_extractor.CLIPTextEncoder",
"keras_cv.models.feature_extractor.QuickGELU",
"keras_cv.models.feature_extractor.ResidualAttention",
],
},
{
Expand All @@ -409,6 +537,30 @@
"keras_cv.models.ImageClassifier.from_preset",
],
},
{
"path": "retinanet",
"title": "The RetinaNet model",
"generate": [
"keras_cv.models.RetinaNet",
"keras_cv.models.RetinaNet.from_preset",
"keras_cv.models.retinanet.PredictionHead"
],
},
{
"path": "stable_diffusion",
"title": "StableDiffusion image-generation model",
"generate": [
"keras_cv.models.StableDiffusion",
"keras_cv.models.StableDiffusionV2",
"keras_cv.models.stable_diffusion.Decoder",
"keras_cv.models.stable_diffusion.DiffusionModel",
"keras_cv.models.stable_diffusion.ImageEncoder",
"keras_cv.models.stable_diffusion.NoiseScheduler",
"keras_cv.models.stable_diffusion.SimpleTokenizer",
"keras_cv.models.stable_diffusion.TextEncoder",
"keras_cv.models.stable_diffusion.TextEncoderV2",
],
},
{
"path": "yolo_v8_detector",
"title": "The YOLOV8Detector model",
Expand Down

0 comments on commit ede83b8

Please sign in to comment.