最佳PyTorch工具集合：AI开发者的实用资源
HorNet
HorNet高阶空间交互
基于递归门控卷积的高效视觉骨干网络
torch-conv-kan
TorchConv KANKolmogorov-Arnold Networks
引入基于Kolmogorov-Arnold表示理论的高效卷积神经网络
text2cinemagraph
Text2CinemagraphStable Diffusion
文本描述生成电影图像的全自动方法
PARE
PARE3D人体姿态估计
三维人体姿态与形状估计的遮挡问题解决方案
V2V-PoseNet_RELEASE
V2V-PoseNet3D手势估计
从单个深度图进行高精度3D手部和人体姿态预测
carefree-learn
PyTorchdeep learning
简化深度学习流程，支持PyTorch高效训练与推理
InSPyReNet
InSPyReNet显著性目标检测
优化显著目标检测的高分辨率图像金字塔网络
3D-BoundingBox
3D Bounding BoxPyTorch
使用深度学习与几何方法，实现高效的3D边界框估计
MiVOS
MiVOS视频对象分割
交互式视频对象分割方法与差异感知融合
3DMPPE_ROOTNET_RELEASE
PyTorch3D姿态估计
单张RGB图像的相机距离感知的3D多人人体姿态估计实现
acezero
ACE0PyTorch
基于增量学习的图像集合场景重建与姿态估计方法
Stylized-ImageNet
ImageNetStylized-ImageNet
介绍如何在卷积神经网络中创建和使用风格化的ImageNet数据集
Mask3D
Mask3D3D实例分割
改进3D语义实例分割 方法，兼容多种数据集
brocolli
brocolliPyTorch
基于Torch FX的PyTorch模型转换和量化工具，支持转换为Caffe和ONNX格式
onnx2torch
onnx2torchONNX
ONNX模型转换至PyTorch的转换器
onnx2tflite
ONNXTFLite
优化ONNX模型到TensorFlow Lite的转换工具，具备高一致性和自动通道对齐功能
pytorch-auto-drive
PytorchAutoDrive语义分割
基于 PyTorch 的分割模型和车道检测模型
YOLOv8-TensorRT
YOLOv8TensorRT
通过TensorRT加速YOLOv8模型，提供在CUDA环境下的快速部署和推理解决方案
AlphaFold3
AlphaFold3深度学习
预测蛋白质相互作用结构的开源工具
model-vs-human
modelvshumanPyTorch
用于评估人类与机器视觉差距的Python工具箱
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号