← all repositories

leondgarse/keras_cv_attention_models

Keras implementation of vision transformers, attention mechanisms, detection models, and diffusion models.

keras_cv_attention_models
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

This repository provides Keras implementations of numerous vision transformer architectures including BEiT, ConvNeXt, EfficientNet, Swin, DINO, and others. It also includes implementations of object detection models (YOLO, EfficientDet), diffusion models (DDPM, Stable Diffusion), CLIP training, and segmentation models. Supports both TensorFlow and PyTorch backends with training pipelines for ImageNet, COCO, and other datasets.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.