keras_cv_attention_models

一站式加载 Keras 主流视觉模型，快速做分类检测分割

keras_cv_attention_models 是一个基于 TensorFlow/Keras 的计算机视觉模型库，汇集了众多主流和前沿的视觉骨干网络，如 BEiT、CoAtNet、ConvNeXt、Swin、MaxViT、MobileViT 等，并支持图像分类、目标检测（YOLO 系列）、分割（Segment Anything）等任务。它解决了 Keras 生态中缺乏统一、易用的先进视觉模型实现的问题，提供了一致的 API 和预训练权重，方便研究者快速对比和部署。核心能力包括：一键加载预训练模型、支持多种注意力机制和高效架构、集成检测与分割工具、提供训练和推理脚本。项目代码结构清晰，兼容 TF2，适合作为视觉任务的基础库。