tensorflow-image-models

将PyTorch图像模型移植到TensorFlow的预训练模型库

TensorFlow 图像模型预训练权重机器学习深度学习 Github 开源项目

tensorflow-image-models是一个将PyTorch图像模型移植到TensorFlow的开源项目。它提供了多种预训练模型,包括ViT、DeiT、ResNet等,可用于图像分类和分割。该项目为开发者提供了简单的API来创建、预处理和保存/加载模型,并支持调整类别数量以适应不同任务。通过这个模型库,研究人员和开发者可以更方便地在TensorFlow中使用先进的图像模型。

访问官网

Github

介绍相关项目

vit-base-patch32-384 - Vision Transformer图像分类模型支持大规模数据训练

GithubHuggingfaceImageNetVision Transformer图像分类开源项目模型深度学习计算机视觉

Vision Transformer（ViT）是一款图像分类模型，采用Transformer编码器架构，通过将图像分割为固定大小patch进行处理。模型在包含1400万张图像的ImageNet-21k数据集完成预训练，并在ImageNet-1k数据集上进行384x384分辨率的微调。提供预训练权重，可直接应用于图像分类或迁移学习任务。

yolov3-tf2 - YOLOv3的TensorFlow实现，目标检测解决方案

GithubTensorFlow 2.0YoloV3开源项目检测训练预训练权重

该项目采用TensorFlow 2.0实现YOLOv3，提供预训练权重、推理示例和迁移学习功能，支持GPU加速、eager模式和图模式训练，并集成absl-py。用户可以方便地安装、训练和进行实时视频检测，同时支持TF模型导出和Serving。

tf_efficientnetv2_s.in21k - EfficientNetV2图像分类模型支持多种计算机视觉应用

EfficientNet-v2GithubHuggingfaceImageNet-21ktimm图像分类开源项目模型特征提取

tf_efficientnetv2_s.in21k是一个基于EfficientNet-v2架构的图像分类模型，在ImageNet-21k数据集上训练。该模型由TensorFlow原始训练，后移植至PyTorch，拥有4820万参数。模型支持图像分类、特征提取和图像嵌入等功能，适用于多种计算机视觉应用场景。通过timm库，开发者可以便捷地加载此预训练模型，实现图像分类、特征图提取或生成图像嵌入等任务。

ViT-Prisma - 视觉变换器和CLIP模型机制解析开源库

GithubVision Transformer图像处理开源库开源项目机器学习解释性神经网络可视化

ViT-Prisma是一个专注于Vision Transformer和CLIP模型的开源机制解析库。它提供logit归因、注意力可视化和激活修补等技术，用于深入分析模型内部机制。该库还包含ViT训练代码和预训练模型，支持ImageNet-1k和dSprites分类任务。ViT-Prisma为视觉模型可解释性研究提供了实用的工具集。

Transfer-Learning-Library - 高效且易用的迁移学习库，支持多种算法和任务

APIGithubPyTorchTLlibTransfer Learning开源项目机器学习

Transfer Learning Library (TLlib) 是一个开源的迁移学习库，基于PyTorch设计，具备高性能和易用性。该库支持多种方法，如域对齐、域转换和半监督学习，适用于分类、回归、目标检测、分割和关键点检测等任务。提供丰富的示例代码和详细文档，并支持pip安装。这是研发新算法或应用现有算法的理想工具，适用于研究和工程实践。

vit_tiny_patch16_384.augreg_in21k_ft_in1k - ViT-Tiny 轻量级视觉转换器模型实现图像分类与特征提取

GithubHuggingfaceImageNetVision Transformertimm图像分类开源项目模型深度学习

ViT-Tiny是一款轻量级视觉转换器模型，专为图像分类和特征提取而设计。该模型在ImageNet-21k上预训练，并在ImageNet-1k上微调，采用了先进的数据增强和正则化技术。模型仅有5.8M参数，能处理384x384尺寸的图像，通过timm库可轻松加载用于推理或进一步微调。ViT-Tiny在保持高性能的同时，大幅降低了计算资源需求，适合各类图像识别应用场景。

vision - TorchVision 计算机视觉库提供数据集模型和图像处理功能

GithubPyTorchtorchvision图像处理开源项目深度学习计算机视觉

TorchVision是PyTorch生态系统的计算机视觉库，提供常用数据集、模型架构和图像变换功能。它支持torch张量和PIL图像后端，具备视频处理能力。该库同时提供Python和C++ API，适用于各种计算机视觉任务。TorchVision版本与PyTorch和Python版本兼容，持续更新以支持最新技术。

efficientnetv2_rw_m.agc_in1k - EfficientNetV2模型：图像分类与多功能特征提取

EfficientNet-v2GithubHuggingfaceImageNet-1ktimm图像分类开源项目模型模型使用

EfficientNetV2是一个在timm库中实现的高效图像分类模型。通过使用以ResNet Strikes Back为基础的训练策略和SGD优化器（带Nesterov动量），结合自适应梯度剪裁，模型在ImageNet-1k数据集上进行训练。这一架构轻量且强大，支持包括图像分类、特征提取和图像嵌入的多种图像处理任务。

maxvit_base_tf_512.in21k_ft_in1k - MaxViT图像分类模型支持多尺寸特征提取和深度学习训练

GithubHuggingfaceImageNetMaxViT人工智能图像分类开源项目模型深度学习

MaxViT是谷歌研究团队开发的图像分类模型，通过ImageNet-21k预训练和ImageNet-1k微调实现。模型集成多轴注意力机制，总参数量119.9M，支持512x512分辨率输入。除图像分类外，还可输出多尺度特征图和嵌入向量，便于迁移至其他视觉任务。模型在ImageNet-1k测试中取得88.20%的分类准确率。

fast-DiT - 改进PyTorch实现的可扩展扩散模型转换器

DiTGithubPyTorchTransformer图像生成开源项目扩散模型

fast-DiT 项目提供了扩散模型转换器（DiT）的改进 PyTorch 实现。该项目包含预训练的类条件 DiT 模型、Hugging Face Space 和 Colab 笔记本，以及优化的训练脚本。通过采用梯度检查点、混合精度训练和 VAE 特征预提取等技术，显著提升了训练速度和内存效率。这一实现为研究人员和开发者提供了探索和应用扩散模型的有力工具。

相关项目

项目侧边栏1

项目侧边栏2

推荐项目

Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手，通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能，支持100+编程语言，与主流编辑器无缝集成，显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台，能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐，Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案，通过智能技术简化视频制作流程。无论是企业宣传还是个人分享，有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持，能够阅读和理解用户上传的文件内容，解析网页信息，并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题，Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术，为商家提供一键生成商品图和营销文案的服务，显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台，让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型，免费访问各种图像生成与AI训练工具，从Stable Diffusion等基础模型开始，轻松实现创新图像生成。体验前沿的AI技术，推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器，它将改变您观看视频的方式！SubCat结合了先进的人工智能技术，为您提供即时视频字幕翻译，无论是本地视频还是网络流媒体，让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台，利用前沿AI技术，为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图，再到文案生成，美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具，简化了选题、文献检索至论文撰写的整个过程。通过简单设定，平台可快速生成高质量论文大纲和全文，配合图表、参考文献等一应俱全，同时提供开题报告和答辩PPT等增值服务，保障数据安全，有效提升写作效率和论文质量。

使用协议隐私政策广告服务

投诉举报邮箱: service@vectorlightyear.com

@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号