Project Icon

pytorch-metric-learning

高级度量学习工具库,适用于深度学习

PyTorch Metric Learning提供多种损失函数、挖掘器和评估工具,支持自定义和无监督学习。该库适用于各种训练和测试需求,配有详细文档和示例,适合高效嵌入空间计算的应用场景。

OfflineRL-Kit - 高效易用的PyTorch离线强化学习库
GithubPyTorch实验管理开源项目模型训练离线强化学习算法库
OfflineRL-Kit是基于PyTorch的离线强化学习库,提供清晰的代码结构和最新算法实现。支持CQL、TD3+BC等多种算法,具备高扩展性和强大的日志系统。该库还支持并行调优,便于研究人员进行实验。相比其他离线强化学习库,OfflineRL-Kit在性能和易用性方面都有显著优势,是离线强化学习研究的有力工具。
ranking - TensorFlow平台上的学习排名技术库
GithubTensorBoardTensorFlow Ranking学习排序开源项目排序模型深度学习
TensorFlow Ranking是一个适用于学习排名(LTR)技术的开源库,基于TensorFlow平台。该库包括点对、成对和列表损失函数,以及各类排名指标如平均倒数排名(MRR)和标准折扣累积增益(NDCG)。它提供了群组评分功能和LambdaLoss的实现,用于优化排名指标。此外,它还支持从偏见反馈数据中进行无偏学习。该库旨在为学术研究和工业应用提供一个开放、便捷的平台。用户可通过教程和演示快速入门,无需复杂安装。
mlgb - 多模型支持的CTR预测和推荐系统库
CTR预测GithubMLGB开源项目推荐系统机器学习深度学习模型
MLGB是一个Python库,集成了50多种CTR预测和推荐系统模型,兼容TensorFlow和PyTorch框架。该库提供简洁的API,方便快速调用复杂模型。通过代码优化,MLGB实现了高效性能,为研究和工程实践提供了多样化的模型选择。
vision3d - PyTorch驱动的开源激光雷达感知库
3D检测GithubLIDAR感知PV-RCNNPyTorchSECOND模型开源项目
Vision 3D是一个基于PyTorch的开源库,专注于激光雷达感知领域。该项目以代码简洁性为核心,便于扩展新模型和数据集。目前已实现SECOND算法,并部分完成PV-RCNN。虽然开发已暂停,但项目仍提供详细文档和示例,支持研究人员和开发者在3D目标检测领域的应用。Vision 3D作为开源工具,欢迎社区贡献,为激光雷达感知研究提供了有价值的资源。
pytorch-fid - 生成对抗网络图像质量评估工具
FIDFréchet Inception DistanceGANsGithubPyTorchTensorflow开源项目
pytorch-fid是一款用于计算生成对抗网络(GAN)样本质量的Fréchet Inception Distance(FID)分数的工具。该工具将官方的Tensorflow实现移植到PyTorch,确保相似的准确性和方便性。用户可以自由选择特征层,适应不同的数据集,还支持GPU加速和保存原始数据集的统计信息,便于进行多模型比较,适合研究和开发高质量图像生成模型。
multi-object-tracker - 利用Python实现多对象跟踪,兼容多种检测器
CentroidTrackerGithubOpenCVTF-MobileNetSSDYOLOv3multi-object tracker开源项目
该项目提供多种基于Python的多对象跟踪算法,包括CentroidTracker、IOUTracker、CentroidKF_Tracker和SORT,支持TF_SSDMobileNetV2、Caffe_SSDMobileNet和YOLOv3等OpenCV对象检测器。安装简便,使用友好,支持GPU加速,适用于视频数据解析和对象追踪。参考项目示例可快速上手,实现精准多对象跟踪。
mlp - 多层感知器实现n-gram语言模型的开源项目
GithubPyTorch多层感知器开源项目神经网络自动微分自然语言模型
该项目基于Bengio等人2003年的论文,实现了多层感知器(MLP)作为n-gram语言模型。项目提供C、NumPy和PyTorch三种实现,展示了从底层操作到高级抽象的不同层次。通过对比,突出了PyTorch在Tensor处理、自动微分和深度学习层构建方面的优势。相比传统n-gram模型,此方法以较少参数实现更低验证损失,但训练成本较高。
Open3D-ML - Open3D 的扩展,用于处理 3D 机器学习任务
3D机器学习GithubOpen3D-MLPyTorchTensorFlow开源项目语义分割
Open3D-ML基于Open3D库,扩展了3D机器学习工具,支持语义点云分割和目标检测等应用。提供预训练模型和训练管道,兼容TensorFlow和PyTorch框架,易于集成到现有项目中。同时,提供数据可视化等通用功能,覆盖多种数据集和算法,提高3D数据处理效率和效果。
Pytorch-NLU - 轻量级NLP工具包 支持文本分类和序列标注
GithubPytorch-NLU序列标注开源项目文本分类自然语言处理预训练模型
Pytorch-NLU是一个轻量级自然语言处理工具包,专注于文本分类、序列标注和文本摘要任务。该工具包支持BERT、ERNIE等多种预训练模型,提供多种损失函数,具有依赖少、代码简洁、注释详细、配置灵活等特点。Pytorch-NLU包含丰富的数据集,使用方式简单,可快速应用于实际NLP项目中。
torchrec - 旨在提供大规模推荐系统所需的常见稀疏性和并行性原语的PyTorch库
CUDAFBGEMMGithubPyTorchTorchRec开源项目推荐系统
TorchRec是一个专为大规模推荐系统设计的PyTorch库,提供稀疏性和并行性解决方案。它支持多种嵌入表分片策略,并能自动优化分片计划。通过流水线训练和优化内核,提高模型性能。还支持量化训练和推理,包含多个验证的模型架构和数据集示例,适用于需要高性能和扩展性的推荐系统项目。
项目侧边栏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号