最佳PyTorch工具集合：AI开发者的实用资源
deep_sort_pytorch
Deep SortPyTorch
使用PyTorch实现的Deep Sort多目标追踪算法
lightly
Lightly自监督学习
简单易用的自监督学习工具，支持自定义骨干模型和分布式训练
imgclsmob
deep learningcomputer vision
深度学习卷积网络的研究与实现，涵盖多种框架和预训练模型
a-PyTorch-Tutorial-to-Object-Detection
PyTorch对象检测
PyTorch物体检测模型教程与实现
DeepRL
DeepRLPyTorch
PyTorch 中深度强化学习算法的模块化实现
practicalAI-cn
practicalAIPyTorch
PyTorch与Google Colab下的机器学习与深度学习实践
machine-learning-book
Machine LearningPyTorch
深入使用PyTorch和Scikit-Learn的机器学习指南
RecBole
RecBole推荐系统
基于Python和PyTorch的推荐系统框架，支持91种算法和43个数据集
pytorch-fid
FIDPyTorch
生成对抗网络图像质量评估工具
mmpretrain
OpenMMLabMMPreTrain
支持多种模型与大规模训练配置的PyTorch开源预训练工具箱
pomegranate
pomegranatePyTorch
Python 中快速、灵活且易于使用的概率建模
hummingbird
Hummingbird机器学习模型
用于将经过训练的传统 ML 模型编译为张量计算的库
fast-reid
FastReIDPyTorch
重识别方法和工具箱
doctr
docTROCR
由深度学习提供支持的无缝、高性能和可访问的库，用于 OCR 相关任务
gpytorch
GPyTorchGaussian process
基于PyTorch实现的灵活高斯过程建模工具
PINTO_model_zoo
PINTO_model_zoo量化
提供多框架神经网络模型转换与量化的开源工具
PerceptualSimilarity
LPIPSBAPPS
利用深度特征提供图像块相似度评估的方法
silero-vad
Silero VAD音频处理
企业级预训练语音活动检测工具，适用于多种平台
kaolin
KaolinPyTorch
提供3D深度学习工具的PyTorch库
SAM-Adapter-PyTorch
SAM-AdapterSegment Anything
提升复杂场景下图像分割效果的开源项目
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号