Project Icon

torch-fidelity

PyTorch生成模型评估指标库

torch-fidelity是一个开源的PyTorch库,提供多种生成模型评估指标的实现。该库支持ISC、FID、KID、PRC和PPL等指标,特点是计算精确、效率高且易于扩展。它通过特征共享和缓存机制优化性能,适用于学术研究和模型训练中的实时评估。torch-fidelity还支持自定义,可适应不同的数据类型和模型结构。

deep-person-reid - 深度学习人员重识别库,支持多GPU训练和跨数据集评估
GithubPyTorchTorchreid多GPU训练开源项目深度学习重识别
Torchreid是一个基于PyTorch的深度学习人员重识别库,支持多GPU训练、图像和视频重识别、端到端训练与评估、多数据集训练和跨数据集评估。它易于准备数据集,支持添加模型、数据集和训练方法,提供预训练模型和高级训练技术,并配备可视化工具。
pytorch-image-models - 全面的PyTorch图像模型集合
GithubPyTorch图像模型开源项目深度学习神经网络计算机视觉
pytorch-image-models是一个综合性PyTorch图像模型库,提供最新计算机视觉模型、预训练权重和训练脚本。库中包含CNN和Transformer等多种架构,支持迁移学习和特征提取。项目不断更新,近期新增MobileNetV4模型并优化现有模型性能。该库为计算机视觉研究和开发提供了丰富的工具和资源。
hub - 开源预训练模型共享与发布平台
CIGithubNetlifyPyTorch Hubhubconf.py开源项目模型提交
PyTorch Hub是一个开源平台,用于发布和共享预训练深度学习模型。开发者可通过简单的PR流程提交模型,平台提供详细的提交指南和模板。PyTorch Hub支持添加hubconf.py文件验证模型功能,并集成了本地测试、CI和网页预览功能,确保模型的正确性和可用性。提交的模型会在24小时内在PyTorch官网展示,促进了AI社区的知识共享和协作。
stable-audio-metrics - 音频生成模型评估指标集合
GPU支持Githubstable-audio-metrics开源项目数据结构音频指标音频生成模型评估
stable-audio-metrics是一个评估音乐和音频生成模型的开源指标集合。它包含基于Openl3的Fréchet距离、基于PaSST的Kullback-Leibler散度和基于CLAP-LAION的CLAP分数。该项目针对长形式全频带立体声生成进行了优化,支持可变长度输入,并提供了详细文档和示例。适用于评估MusicCaps、AudioCaps和Song Describer等数据集的音频生成质量。
ffcv-imagenet - 高效ImageNet训练框架提升模型性能
GithubImageNetPyTorchResNetffcv开源项目深度学习
ffcv-imagenet是一个高效的ImageNet训练框架,采用单文件PyTorch脚本实现。该项目能在标准方法1/10的时间内达到相同精度,支持多GPU并行和多模型同时训练。框架提供丰富的配置选项,结合FFCV数据加载和优化训练流程,使研究人员能更快迭代实验并获得高质量模型。项目还包含多种预设配置,适用于不同的训练需求和硬件环境。
Person_reID_baseline_pytorch - 小巧、友好、强大的 pytorch 工具
GPUGithubPytorch ReID对象识别开源项目教程深度学习
Pytorch ReID是一个高效且易用的对象重识别代码库,支持多种先进的模型与损失函数,如ResNet、Swin Transformer和Circle Loss。该项目自2017年起持续更新,拥有详细的教程与训练评估功能,性能在多篇顶级会议论文中得到验证,适合各种经验水平的用户使用。
torchdyn - PyTorch数值深度学习库,支持微分方程和数值方法
GithubPyTorchTorchdyn开源项目微分方程数值方法深度学习
Torchdyn是一个专注于数值深度学习的PyTorch库,涵盖微分方程、积分变换和数值方法。它提供便捷的工具和层,用于构建神经微分方程和复合模型,并支持GPU加速和多种数值方法。该库与PyTorch和pytorch-lightning高度集成,使得用户能够快速上手,推进研究和应用。
pytorch_tabular - 表格数据深度学习的简易化解决方案
GithubPyTorchPyTorch Lightning开源项目标签数据模型定制深度学习
PyTorch Tabular是一个开源库,旨在简化表格数据的深度学习应用。该库具有低阻力易用性、易于定制和可扩展性,基于PyTorch和PyTorch Lightning构建。提供如TabNet、NODE和GATE等多种模型选择,适用于研究和实际应用。详细的文档和简单的安装流程使用户能够快速上手和自定义模型,提高机器学习任务的效率和性能。
FAST-VQA-and-FasterVQA - 开源高效视频质量评估框架
FAST-VQAFasterVQAGithub开源项目机器学习深度学习视频质量评估
FAST-VQA和FasterVQA是端到端视频质量评估的开源工具箱,提供高效的评估模型。FasterVQA作为FAST-VQA的改进版,在保持相似性能的同时速度提升4倍。这些模型在多个数据集上达到最先进水平。项目采用模块化架构,支持灵活的空间和时间采样方法及多种网络结构。研究者可进行模型训练、测试,并在小型数据集上微调。
flashtorch - 基于PyTorch的神经网络可视化工具
FlashTorchGithubPyTorch可视化开源项目特征可视化神经网络
FlashTorch是基于PyTorch的神经网络可视化工具,通过简单的接口实现特征可视化技术,如显著性图和激活最大化。该工具兼容torchvision预训练模型和自定义PyTorch模型,有助于研究人员和开发者理解、解释及优化神经网络的内部工作机制。FlashTorch仅需几行代码即可应用,为深入分析神经网络提供了便捷途径。
项目侧边栏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

稿定AI

稿定设计 是一个多功能的在线设计和创意平台,提供广泛的设计工具和资源,以满足不同用户的需求。从专业的图形设计师到普通用户,无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑,稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合,帮助用户轻松实现创意设计。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号