最佳PyTorch工具集合：AI开发者的实用资源
pygod
PyGODPython
图形异常检测的Python库
torchdyn
TorchdynPyTorch
PyTorch数值深度学习库，支持微分方程和数值方法
nerfacc
NerfAccPyTorch
NeRF训练与推理的PyTorch加速工具箱
torchtyping
torchtypingtype annotations
张量类型注解工具，支持形状和数据类型检查
kaolin-wisp
NVIDIA Kaolin WispPyTorch
用于神经场研究的PyTorch库
bindsnet
BindsNET脉冲神经网络
模拟尖峰神经网络的生物启发机器学习算法
pytorch-toolbelt
PyTorch深度学习
专为PyTorch设计的Python库，提供高效研发和Kaggle竞赛所需的工具集
Det3D
Det3D3D对象检测
提供多数据集和算法支持的3D目标检测工具箱
vocal-remover
vocal-remover深度学习
使用深度学习的开源伴奏提取工具
3d-ken-burns
3d-ken-burnsPyTorch
从单幅图像生成3D Ken Burns效果
gen-efficientnet-pytorch
EfficientNetPyTorch
泛型EfficientNet和其它高效PyTorch模型的实现
pytorch_scatter
PyTorch数据处理
优化分散操作的 PyTorch 扩展库
theseus
Theseus非线性优化
构建适用于机器人和视觉应用的定制非线性优化层
torchsde
PyTorchstochastic differential equation
提供GPU支持的随机微分方程求解器
tnt
TNTPyTorch
PyTorch训练库，简化和优化模型训练过程
opacus
OpacusPyTorch
简化PyTorch差分隐私训练流程
IQA-PyTorch
PyTorchIQA
纯Python和PyTorch图像质量评估工具箱
EasyCV
EasyCVPyTorch
基于PyTorch的全能计算机视觉工具箱，支持自监督学习和Transformer模型
tutorials
MONAIPyTorch
涵盖2D和3D分类、分割、回归及配准任务实例MONAI教程
lightning-flash
lightning-flashPyTorch
跨数据领域和任务的AI模型训练与处理解决方案
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号