Skip to content

Commit d2da5ea

Browse files
sachinprasadhssachinprasadhs
authored
Update Models API for Keras CV (#1798)
* Update Models API * Fix path
1 parent c99e77b commit d2da5ea

File tree

1 file changed

+183
-41
lines changed

1 file changed

+183
-41
lines changed

‎scripts/cv_api_master.py

+183-41
Original file line numberDiff line numberDiff line change
@@ -306,6 +306,98 @@
306306
"title": "Backbones",
307307
"toc": True,
308308
"children": [
309+
{
310+
"path": "csp_darknet",
311+
"title": "CSPDarkNet backbones",
312+
"generate": [
313+
"keras_cv.models.CSPDarkNetBackbone",
314+
"keras_cv.models.CSPDarkNetBackbone.from_preset",
315+
"keras_cv.models.CSPDarkNetTinyBackbone",
316+
"keras_cv.models.CSPDarkNetSBackbone",
317+
"keras_cv.models.CSPDarkNetMBackbone",
318+
"keras_cv.models.CSPDarkNetLBackbone",
319+
"keras_cv.models.CSPDarkNetXLBackbone",
320+
],
321+
},
322+
{
323+
"path": "densenet",
324+
"title": "DenseNet backbones",
325+
"generate": [
326+
"keras_cv.models.DenseNetBackbone",
327+
"keras_cv.models.DenseNetBackbone.from_preset",
328+
"keras_cv.models.DenseNet121Backbone",
329+
"keras_cv.models.DenseNet169Backbone",
330+
"keras_cv.models.DenseNet201Backbone",
331+
],
332+
},
333+
{
334+
"path": "efficientnet_v1",
335+
"title": "EfficientNetV1 models",
336+
"generate": [
337+
"keras_cv.models.EfficientNetV1Backbone",
338+
"keras_cv.models.EfficientNetV1Backbone.from_preset",
339+
"keras_cv.models.EfficientNetV1B0Backbone",
340+
"keras_cv.models.EfficientNetV1B1Backbone",
341+
"keras_cv.models.EfficientNetV1B2Backbone",
342+
"keras_cv.models.EfficientNetV1B3Backbone",
343+
"keras_cv.models.EfficientNetV1B4Backbone",
344+
"keras_cv.models.EfficientNetV1B5Backbone",
345+
"keras_cv.models.EfficientNetV1B6Backbone",
346+
"keras_cv.models.EfficientNetV1B7Backbone",
347+
],
348+
},
349+
{
350+
"path": "efficientnet_v2",
351+
"title": "EfficientNetV2 models",
352+
"generate": [
353+
"keras_cv.models.EfficientNetV2Backbone",
354+
"keras_cv.models.EfficientNetV2Backbone.from_preset",
355+
"keras_cv.models.EfficientNetV2B0Backbone",
356+
"keras_cv.models.EfficientNetV2B1Backbone",
357+
"keras_cv.models.EfficientNetV2B2Backbone",
358+
"keras_cv.models.EfficientNetV2B3Backbone",
359+
"keras_cv.models.EfficientNetV2SBackbone",
360+
"keras_cv.models.EfficientNetV2MBackbone",
361+
"keras_cv.models.EfficientNetV2LBackbone",
362+
],
363+
},
364+
{
365+
"path": "efficientnet_lite",
366+
"title": "EfficientNet Lite backbones",
367+
"generate": [
368+
"keras_cv.models.EfficientNetLiteBackbone",
369+
"keras_cv.models.EfficientNetLiteBackbone.from_preset",
370+
"keras_cv.models.EfficientNetLiteB0Backbone",
371+
"keras_cv.models.EfficientNetLiteB1Backbone",
372+
"keras_cv.models.EfficientNetLiteB2Backbone",
373+
"keras_cv.models.EfficientNetLiteB3Backbone",
374+
"keras_cv.models.EfficientNetLiteB4Backbone",
375+
],
376+
},
377+
{
378+
"path": "mix_transformer",
379+
"title": "MixTransformer backbones",
380+
"generate": [
381+
"keras_cv.models.MiTBackbone",
382+
"keras_cv.models.MiTBackbone.from_preset",
383+
"keras_cv.models.MiTB0Backbone",
384+
"keras_cv.models.MiTB1Backbone",
385+
"keras_cv.models.MiTB2Backbone",
386+
"keras_cv.models.MiTB3Backbone",
387+
"keras_cv.models.MiTB4Backbone",
388+
"keras_cv.models.MiTB5Backbone",
389+
],
390+
},
391+
{
392+
"path": "mobilenet_v3",
393+
"title": "MobileNetV3 backbones",
394+
"generate": [
395+
"keras_cv.models.MobileNetV3Backbone",
396+
"keras_cv.models.MobileNetV3Backbone.from_preset",
397+
"keras_cv.models.MobileNetV3SmallBackbone",
398+
"keras_cv.models.MobileNetV3LargeBackbone",
399+
],
400+
},
309401
{
310402
"path": "resnet_v1",
311403
"title": "ResNetV1 backbones",
@@ -332,50 +424,31 @@
332424
"keras_cv.models.ResNet152V2Backbone",
333425
],
334426
},
335-
# {
336-
# "path": "csp_darknet",
337-
# "title": "CSPDarkNet backbones",
338-
# "generate": [
339-
# "keras_cv.models.CSPDarkNetBackbone",
340-
# "keras_cv.models.CSPDarkNetBackbone.from_preset",
341-
# "keras_cv.models.CSPDarkNetTinyBackbone",
342-
# "keras_cv.models.CSPDarkNetSBackbone",
343-
# "keras_cv.models.CSPDarkNetMBackbone",
344-
# "keras_cv.models.CSPDarkNetLBackbone",
345-
# "keras_cv.models.CSPDarkNetXLBackbone",
346-
# ],
347-
# },
348427
{
349-
"path": "yolo_v8",
350-
"title": "YOLOV8 backbones",
428+
"path": "vgg16",
429+
"title": "VGG16 backbones",
351430
"generate": [
352-
"keras_cv.models.YOLOV8Backbone",
353-
"keras_cv.models.YOLOV8Backbone.from_preset",
431+
"keras_cv.models.VGG16Backbone",
432+
"keras_cv.models.VGG16Backbone.from_preset",
354433
],
355434
},
356435
{
357-
"path": "mobilenetv3",
358-
"title": "MobileNetV3 backbones",
436+
"path": "vitdet",
437+
"title": "ViTDet backbones",
359438
"generate": [
360-
"keras_cv.models.MobileNetV3Backbone",
361-
"keras_cv.models.MobileNetV3Backbone.from_preset",
362-
"keras_cv.models.MobileNetV3SmallBackbone",
363-
"keras_cv.models.MobileNetV3LargeBackbone",
439+
"keras_cv.models.ViTDetBackbone",
440+
"keras_cv.models.ViTDetBackbone.from_preset",
441+
"keras_cv.models.ViTDetBBackbone",
442+
"keras_cv.models.ViTDetLBackbone",
443+
"keras_cv.models.ViTDetHBackbone",
364444
],
365445
},
366446
{
367-
"path": "efficientnetv2",
368-
"title": "EfficientNetV2 models",
447+
"path": "yolo_v8",
448+
"title": "YOLOV8 backbones",
369449
"generate": [
370-
"keras_cv.models.EfficientNetV2Backbone",
371-
"keras_cv.models.EfficientNetV2Backbone.from_preset",
372-
"keras_cv.models.EfficientNetV2B0Backbone",
373-
"keras_cv.models.EfficientNetV2B1Backbone",
374-
"keras_cv.models.EfficientNetV2B2Backbone",
375-
"keras_cv.models.EfficientNetV2B3Backbone",
376-
"keras_cv.models.EfficientNetV2SBackbone",
377-
"keras_cv.models.EfficientNetV2MBackbone",
378-
"keras_cv.models.EfficientNetV2LBackbone",
450+
"keras_cv.models.YOLOV8Backbone",
451+
"keras_cv.models.YOLOV8Backbone.from_preset",
379452
],
380453
},
381454
],
@@ -387,18 +460,63 @@
387460
"toc": True,
388461
"children": [
389462
{
390-
"path": "stable_diffusion",
391-
"title": "StableDiffusion image-generation model",
463+
"path": "basnet_segmentation",
464+
"title": "BASNet Segmentation",
392465
"generate": [
393-
"keras_cv.models.StableDiffusion",
466+
"keras_cv.models.BASNet",
467+
"keras_cv.models.BASNet.from_preset",
394468
],
395469
},
396470
{
397-
"path": "retinanet",
398-
"title": "The RetinaNet model",
471+
"path": "deeplab_v3_segmentation",
472+
"title": "DeepLabV3Plus Segmentation",
399473
"generate": [
400-
"keras_cv.models.RetinaNet",
401-
"keras_cv.models.RetinaNet.from_preset",
474+
"keras_cv.models.DeepLabV3Plus",
475+
"keras_cv.models.DeepLabV3Plus.from_preset",
476+
],
477+
},
478+
{
479+
"path": "segformer_segmentation",
480+
"title": "SegFormer Segmentation",
481+
"generate": [
482+
"keras_cv.models.SegFormer",
483+
"keras_cv.models.SegFormer.from_preset",
484+
"keras_cv.models.SegFormerB0",
485+
"keras_cv.models.SegFormerB1",
486+
"keras_cv.models.SegFormerB2",
487+
"keras_cv.models.SegFormerB3",
488+
"keras_cv.models.SegFormerB4",
489+
"keras_cv.models.SegFormerB5",
490+
],
491+
},
492+
{
493+
"path": "segment_anything",
494+
"title": "Segment Anything",
495+
"generate": [
496+
"keras_cv.models.SegmentAnythingModel",
497+
"keras_cv.models.SegmentAnythingModel.from_preset",
498+
"keras_cv.models.SAMMaskDecoder",
499+
"keras_cv.models.SAMPromptEncoder",
500+
"keras_cv.models.TwoWayTransformer",
501+
"keras_cv.layers.MultiHeadAttentionWithDownsampling",
502+
"keras_cv.layers.TwoWayMultiHeadAttention",
503+
"keras_cv.layers.RandomFrequencyPositionalEmbeddings",
504+
],
505+
},
506+
{
507+
"path": "feature_extractor",
508+
"title": "CLIP Feature extractor",
509+
"generate": [
510+
"keras_cv.models.CLIP",
511+
"keras_cv.models.CLIP.from_preset",
512+
"keras_cv.models.feature_extractor.CLIPAttention",
513+
"keras_cv.models.feature_extractor.CLIPEncoder",
514+
"keras_cv.models.feature_extractor.CLIPImageEncoder",
515+
"keras_cv.models.feature_extractor.CLIPPatchingAndEmbedding",
516+
"keras_cv.models.feature_extractor.CLIPProcessor",
517+
"keras_cv.models.feature_extractor.CLIPTextEncoder",
518+
"keras_cv.models.feature_extractor.QuickGELU",
519+
"keras_cv.models.feature_extractor.ResidualAttention",
402520
],
403521
},
404522
{
@@ -409,6 +527,30 @@
409527
"keras_cv.models.ImageClassifier.from_preset",
410528
],
411529
},
530+
{
531+
"path": "retinanet",
532+
"title": "The RetinaNet model",
533+
"generate": [
534+
"keras_cv.models.RetinaNet",
535+
"keras_cv.models.RetinaNet.from_preset",
536+
"keras_cv.models.retinanet.PredictionHead"
537+
],
538+
},
539+
{
540+
"path": "stable_diffusion",
541+
"title": "StableDiffusion image-generation model",
542+
"generate": [
543+
"keras_cv.models.StableDiffusion",
544+
"keras_cv.models.StableDiffusionV2",
545+
"keras_cv.models.stable_diffusion.Decoder",
546+
"keras_cv.models.stable_diffusion.DiffusionModel",
547+
"keras_cv.models.stable_diffusion.ImageEncoder",
548+
"keras_cv.models.stable_diffusion.NoiseScheduler",
549+
"keras_cv.models.stable_diffusion.SimpleTokenizer",
550+
"keras_cv.models.stable_diffusion.TextEncoder",
551+
"keras_cv.models.stable_diffusion.TextEncoderV2",
552+
],
553+
},
412554
{
413555
"path": "yolo_v8_detector",
414556
"title": "The YOLOV8Detector model",

0 commit comments

Comments
 (0)