Pytorch资源大全
目录
Pytorch及相关库
- pytorch:Python中的张量和动态神经网络,具有强大的GPU加速功能。
NLP和语音处理:
- pytorch text:火炬文本相关内容。
- pytorch-seq2seq:PyTorch中实现的序列到序列(seq2seq)模型的框架。
- anuvada:使用PyTorch进行NLP的可解释模型。
- audio:用于pytorch的简单音频I / O.
- loop:一种跨多个扬声器生成语音的方法
- fairseq-py:用Python编写的Facebook AI研究序列到序列工具包。
- speech:PyTorch ASR实施。
- OpenNMT-py:PyTorch http://opennmt.net中的开源神经机器翻译
- neuralcoref:基于神经网络和spaCy huggingface.co/coref的最先进的共指消解
- sentiment-discovery:大规模的无监督语言建模,用于稳健的情绪分类。
- MUSE:多语言无监督或监督词嵌入的库
- nmtpytorch:PyTorch中的神经机器翻译框架。
- pytorch-wavenet:WaveNet的快速生成实现
- Tacotron-pytorch:Tacotron:走向端到端语音合成。
- AllenNLP:一个基于PyTorch构建的开源NLP研究库。
- PyTorch-NLP:PyTorch的文本实用程序和数据集pytorchnlp.readthedocs.io
- quick-nlp:基于FastAI的Pytorch NLP库。
- TTS:深入学习Text2Speech
- LASER:语言不可知的能力表征
- pyannote-audio:用于扬声器分类的神经构建块:语音活动检测,扬声器变化检测,扬声器嵌入
- gensen:通过大规模多任务学习学习通用分布式句子表示。
- translate:翻译 - PyTorch语言库。
- espnet:端到端语音处理工具包espnet.github.io/espnet
- pythia:Visual Question Answering的软件套件
- 无监督的MT:基于短语和神经无监督的机器翻译。
- jiant:jiant句子表示学习工具包。
计算机视觉:
- pytorch vision:特定于Computer Vision的数据集,转换和模型。
- pt-styletransfer:神经风格转移作为PyTorch中的一个类。
- OpenFacePytorch:使用OpenFace的nn4.small2.v1.t7模型的PyTorch模块
- img_classification_pk_pytorch:快速将您的图像分类模型与最先进的模型(如DenseNet,ResNet,......)进行比较
- SparseConvNet:子流形稀疏卷积网络。
- Convolution_LSTM_pytorch:多层卷积LSTM模块
- face-alignment:? 使用pytorch adrianbulat.com构建2D和3D Face对齐库
- pytorch-semantic-segmentation:用于语义分割的PyTorch。
- RoIAlign.pytorch:这是RoIAlign的PyTorch版本。此实现基于crop_and_resize,并支持CPU和GPU上的前向和后向。
- pytorch-cnn-finetune:使用PyTorch微调预训练卷积神经网络。
- detectorch:检测器 - PyTorch的检测器
- Augmentor:用于机器学习的Python图像增强库。http://augmentor.readthedocs.io
- s2cnn:该库包含用于球形信号的SO(3)等变CNN的PyTorch实现(例如全向摄像机,地球上的信号)s
概率/生成库:
- ptstat:PyTorch中的概率编程和统计推断
- pyro:使用Python和PyTorch http://pyro.ai进行深度通用概率编程
- probtorch:Probabilistic Torch是扩展PyTorch的深层生成模型的库。
- paysage:python / pytorch中的无监督学习和生成模型。
- pyvarinf:Python包,便于使用贝叶斯深度学习方法和PyTorch的变分推理。
- pyprob:基于PyTorch的概率编程和推理编译库。
- mia:用于运行针对ML模型的成员资格推断攻击的库。
其他库:
- pytorch extras:pytorch的一些额外功能。
- functional zoo:与lua torch不同,PyTorch在其核心中具有autograd,因此不需要使用torch.nn模块的模块化结构,可以轻松分配所需的变量并编写利用它们的函数,这有时更方便。此repo包含此功能方式的模型定义,某些模型具有预训练权重。
- torch-sampling:该软件包提供一组转换和数据结构,用于从内存或内存不足数据中进行采样。
- torchcraft-py:TorchCraft的 Python包装器,是Torch和星际争霸之间用于人工智能研究的桥梁。
- aorun:Aorun打算和PyTorch一起成为Keras的后端。
- logger:用于实验的简单记录器。
- PyTorch-docset:PyTorch docset!与Dash,Zeal,Velocity或LovelyDocs一起使用。
- convert_torch_to_pytorch:将火炬t7模型转换为pytorch模型和源。
- pretrained-models.pytorch:这个回购的目的是帮助重现研究论文的结果。
- pytorch_fft:用于FFT的PyTorch包装器
- caffe_to_torch_to_pytorch
- pytorch-extension:这是PyTorch的CUDA扩展,它计算两个张量的Hadamard积。
- tensorboard-pytorch:该模块以张量板格式保存PyTorch张量以供检查。目前支持张量板中的标量,图像,音频,直方图功能。
- gpytorch:GPyTorch是一个高斯过程库,使用PyTorch实现。它专为轻松创建灵活的模块化高斯过程模型而设计,因此您无需成为使用GP的专家。
- spotlight:使用PyTorch的深度推荐模型。
- pytorch-cns:使用PyTorch进行压缩网络搜索
- pyinn:CuPy融合PyTorch神经网络运算
- inferno:PyTorch周围的实用程序库
- pytorch-fitmodule:PyTorch模块的超简单拟合方法
- inferno-sklearn:包含pytorch的scikit-learn兼容神经网络库。
- pytorch-caffe-darknet-convert:pytorch,caffe prototxt / weights和darknet cfg / weights之间的转换
- pytorch2caffe:将PyTorch模型转换为Caffemodel
- pytorch-tools:PyTorch的工具
- sru:训练RNN与CNN一样快(arxiv.org/abs/1709.02755)
- torch2coreml:Torch7 - > CoreML
- PyTorch编码:PyTorch深度纹理编码网络http://hangzh.com/PyTorch-Encoding
- pytorch-ctc:PyTorch-CTC是PyTorch的CTC(连接主义时间分类)波束搜索解码的实现。C ++代码从TensorFlow中大量借用,并进行了一些改进以增加灵活性。
- candlegp:Pytorch中的高斯过程。
- dpwa:通过Pair-Wise Averaging进行分布式学习。
- dni-pytorch:使用PyTorch的合成梯度解耦神经接口。
- skorch:包含pytorch的scikit-learn兼容神经网络库
- ignite:Ignite是一个高级库,可以帮助在PyTorch中训练神经网络。
- Arnold:阿诺德 - DOOM特工
- pytorch-mcn:将模型从MatConvNet转换为PyTorch
- simple-faster-rcnn-pytorch:简化R-CNN的简化实施,具有竞争力的性能。
- generative_zoo:generative_zoo是一个存储库,它在PyTorch中提供了一些生成模型的工作实现。
- pytorchviz:一个小包,用于创建PyTorch执行图的可视化。
- cogitare:Cogitare - Python中的一个现代,快速,模块化的深度学习和机器学习框架。
- pydlt:基于PyTorch的深度学习工具箱
- semi-supervised-pytorch:在PyTorch中实现不同的基于VAE的半监督和生成模型。
- pytorch_cluster:优化图集群算法的PyTorch扩展库。
- neural-assembly-compiler:基于自适应神经编译的pyTorch神经汇编编译器。
- caffemodel2pytorch:将Caffe模型转换为PyTorch。
- extension-cpp:PyTorch中的C ++扩展
- pytoune:类似Keras的框架和PyTorch实用程序
- jetson-reinforcement:用于NVIDIA Jetson TX1 / TX2的深层强化学习库,包括PyTorch,OpenAI Gym和Gazebo机器人模拟器。
- matchbox:在各个示例的级别编写PyTorch代码,然后在minibatches上高效运行它。
- torch-two-sample:用于双样本测试的PyTorch库
- pytorch-摘要:在PyTorch模型汇总类似于
model.summary()
在Keras - mpl.pytorch:Pytorch实现MaxPoolingLoss。
- scVI-dev:PyTorch中scVI项目的开发分支
- apex:实验性PyTorch扩展(稍后将弃用)
- ELF:ELF:游戏研究的平台。
- Torchlite:在(不仅仅是)Pytorch之上的高级库
- joint-vae:Pytorch实施的JointVAE,一个解开连续和离散变异因子star2的框架
- SLM-Lab:PyTorch中的模块化深度强化学习框架。
- bindsnet:一个Python包,用于使用PyTorch在CPU或GPU上模拟尖峰神经网络(SNN)
- pro_gan_pytorch:作为PyTorch nn.Module的扩展实现的ProGAN包
- pytorch_geometric:PyTorch的几何深度学习扩展库
- torchplus:在PyTorch模块上实现+运算符,返回序列。
- lagom:lagom:一种轻型PyTorch基础设施,可快速构建强化学习算法原型。
- torchbearer:火炬手:使用PyTorch的研究人员的模型训练库。
- pytorch-maml-rl:在Pytorch中使用模型不可知元学习进行强化学习。
- NALU:来自神经算术逻辑单元的NAC / NALU的基本pytorch实现,作者:trask et.al arxiv.org/pdf/1808.00508.pdf
- QuCumber:神经网络多体波函数重建
- magnet:自学的深度学习项目http://magnet-dl.readthedocs.io/
- opencv_transforms:OpenCV实现Torchvision的图像增强
- fastai:fast.ai深度学习库,课程和教程
- pytorch-dense-correspondence:“密集对象网络的代码:通过机器人操作学习密集视觉对象描述符”arxiv.org/pdf/1806.08756.pdf
- colorization-pytorch:交互式深色着色的PyTorch重新实现richzhang.github.io/ideepcolor
- beauty-net:PyTorch的一个简单,灵活且可扩展的模板。很美丽。
- OpenChem:OpenChem:用于计算化学和药物设计研究的深度学习工具包mariewelt.github.io/OpenChem
- torchani:PyTorch aiqm.github.io/torchani上精确的神经网络潜力
- PyTorch-LBFGS:L-BFGS的PyTorch实现。
- gpytorch:PyTorch中高斯过程的高效和模块化实现。
- hessian:在pytorch中的hessian。
- vel:深度学习研究中的速度。
- nonechucks:跳过PyTorch DataLoader中的坏项,使用Transforms as Filters等等!
教程和示例
- 实用的Pytorch:解释不同RNN模型的教程
- DeepLearningForNLPInPytorch:关于深度学习的IPython Notebook教程,重点是自然语言处理。
- pytorch-tutorial:研究人员用pytorch学习深度学习的教程。
- pytorch-exercises:pytorch-exercises集合。
- pytorch教程:各种pytorch教程。
- pytorch examples:展示使用pytorch的示例的存储库
- pytorch practice:pytorch上的一些示例脚本。
- pytorch迷你教程:PyTorch的最小教程改编自Alec Radford的Theano教程。
- pytorch文本分类:Pytorch中基于CNN的文本分类的简单实现
- 猫与狗:pytorch网络微调的例子,对于kaggle比赛Dogs vs. Cats Redux:Kernels Edition。目前在排行榜上排名第27(0.05074)。
- convnet:这是深度卷积网络在各种数据集(ImageNet,Cifar10,Cifar100,MNIST)上的完整培训示例。
- pytorch-generative-adversarial-networks:使用PyTorch的简单生成对抗网络(GAN)。
- pytorch容器:该存储库旨在通过提供Torch表层的PyTorch实现列表,帮助前Torchies更加无缝地过渡到PyTorch的“无容器”世界。
- pytorch中的T-SNE:pytorch中的 t-SNE实验
- AAE_pytorch:Adversarial Autoencoders(与Pytorch合作)。
- Kind_PyTorch_Tutorial:适合初学者的PyTorch教程。
- pytorch-poetry-gen:基于pytorch的char-RNN。
- pytorch-REINFORCE:REINFORCE的PyTorch实现,这个repo支持OpenAI gym中的连续和离散环境。
- PyTorch-Tutorial:轻松快速地构建您的神经网络 https://morvanzhou.github.io/tutorials/
- pytorch-intro:几个脚本,用于说明如何在PyTorch中执行CNN和RNN
- pytorch-classification:CIFAR-10/100和ImageNet上图像分类任务的统一框架。
- pytorch_notebooks - hardmaru:在NumPy和PyTorch中创建的随机教程。
- pytorch_tutoria-quick:快速PyTorch介绍和教程。目标是计算机视觉,图形和机器学习研究人员渴望尝试新的框架。
- Pytorch_fine_tuning_Tutorial:一个关于在PyTorch中执行微调或转移学习的简短教程。
- pytorch_exercises:pytorch-exercises
- 交通标志检测:nyu-cv-fall-2017示例
- mss_pytorch:通过循环推理和跳过过滤连接进行语音分离 - PyTorch实现。演示:js-mim.github.io/mss_pytorch
- DeepNLP-models-Pytorch Pytorch在cs-224n中实现各种Deep NLP模型(Stanford Univ:NLP with Deep Learning)
- Mila入门教程:为欢迎MILA新生提供各种教程。
- pytorch.rl.learning:用于学习使用PyTorch进行强化学习。
- minimal-seq2seq:在PyTorch中注意神经机器翻译的最小Seq2Seq模型
- tensorly-notebooks:TensorLy中的 Tensor方法tensorly.github.io/dev
- pytorch_bits:时间序列预测相关的例子。
- skip-thoughts:在PyTorch中实现Skip-Thought Vectors。
- video-caption-pytorch:用于视频字幕的pytorch代码。
- Capsule-Network-Tutorial:Pytorch易于关注的胶囊网络教程。
- 学习代码 - 深度学习与pytorch:这是“使用PyTorch学习深度学习”一书的代码item.jd.com/17915495606.html
- RL-Adventure:Pytorch易于遵循的深入Q学习教程,带有干净的可读代码。
- accelerated_dl_pytorch:在亚特兰大二世的Jupyter Day与PyTorch加速深度学习。
- RL-Adventure-2:PyTorch4教程:演员评论家/近端政策优化/宏碁/ ddpg /双人决斗ddpg /软演员评论家/生成对抗模仿学习/后见之明体验重播
- 50行代码中的生成对抗网络(GAN)(PyTorch)
- 对抗性-自动编码与 - pytorch
- 使用pytorch转移学习
- 如何对实施-A-YOLO对象检测器功能于pytorch
- pytorch换推荐器-101
- pytorch换numpy的用户
- PyTorch教程:PyTorch中文教程。
- grokking-pytorch:Hitchiker的PyTorch指南
- PyTorch-Deep-Learning-Minicourse:使用PyTorch进行深度学习的微创。
- pytorch-custom-dataset-examples:PyTorch的一些自定义数据集示例
- 基于序列的推荐者的乘法LSTM
- deeplearning.ai-pytorch:课程深度学习(deeplearning.ai)专业化的PyTorch实现。
- MNIST_Pytorch_python_and_capi:这是一个如何在Python中训练MNIST网络并使用pytorch 1.0在c ++中运行它的示例
论文的实现
- google_evolution:这实现了Esteban Real等人大规模演化图像分类器的结果网络之一。
- pyscatwave:使用CuPy / PyTorch进行快速散射变换,请阅读此处的论文
- scalingscattering:扩展散射变换:深度混合网络。
- deep-auto-punctuation:按字符逐个学习的自动标点符号的pytorch实现。
- Realtime_Multi-Person_Pose_Estimation:这是Realtime_Multi-Person_Pose_Estimation的pytorch版本,原始代码在这里。
- PyTorch-value-iteration-networks:价值迭代网络(NIPS '16)论文的PyTorch实现
- pytorch_Highway:在pytorch中实施的公路网。
- pytorch_NEG_loss:在pytorch中实现的NEG丢失。
- pytorch_RVAE:循环变分自动编码器,用于生成在pytorch中实现的顺序数据。
- pytorch_TDNN:在pytorch中实现的时间延迟NN。
- eve.pytorch:Eve Optimizer的一个实现,在具有反馈的Imploving Stochastic Gradient Descent中提出,Koushik和Hayashi,2016。
- e2e-model-learning:基于任务的端到端模型学习。
- pix2pix-pytorch:PyTorch实现“使用条件对抗网络的图像到图像的翻译”。
- Single Shot MultiBox Detector:单次多盒检测器的PyTorch实现。
- DiscoGAN:PyTorch实施“学习发现与生成对抗网络的跨域关系”
- official DiscoGAN implementation:“学习发现与生成性对抗网络的跨域关系”的官方实施。
- pytorch-es:这是Evolution Strategies的PyTorch实现。
- piwise:使用pytorch对VOC2012数据集进行像素分割。
- pytorch-dqn:pytorch中的深度Q-Learning网络。
- neuraltalk2-pytorch:pytorch中的图像字幕模型(with_finetune分支中的finetunable cnn)
- vnet.pytorch:V-Net的Pytorch实现:用于体积医学图像分割的完全卷积神经网络。
- pytorch-fcn:完全卷积网络的PyTorch实现。
- WideResNets:PyTorch中实现的CIFAR10 / 100的WideResNets。这种实现需要的GPU内存少于官方Torch实现所需的内存:https://github.com/szagoruyko/wide-residual-networks。
- pytorch_highway_networks:在PyTorch中实现的高速公路网络。
- pytorch-NeuCom:Pytorch实现了DeepMind的可微分神经计算机论文。
- captionGen:使用PyTorch为图像生成标题。
- AnimeGAN:生成对抗网络的简单PyTorch实现,专注于动漫人脸绘图。
- Cnn-text classification:这是PyTorch中用于句子分类论文的Kim's卷积神经网络的实现。
- deepspeech2:使用百度Warp-CTC实现DeepSpeech2。创建基于DeepSpeech2架构的网络,使用CTC激活功能进行培训。
- seq2seq:此存储库包含PyTorch中的Sequence to Sequence(Seq2Seq)模型的实现
- PyTorch中的Asynchronous Advantage Actor-Critic:这是在深度强化学习的异步方法中描述的A3C的PyTorch实现。由于PyTorch有一个简单的方法来控制多进程内的共享内存,我们可以轻松实现像A3C这样的异步方法。
- densenet:这是DenseNet-BC架构的PyTorch实现,如G. Huang,Z。Liu,K。Weinberger和L. van der Maaten的文章Densely Connected Convolutional Networks中所述。这个实现的CIFAR-10 +错误率为4.77,其中100层DenseNet-BC的增长率为12.他们的官方实现和许多其他第三方实现的链接可以在GitHub上的liuzhuang13 / DenseNet repo中找到。
- nninit:PyTorch nn.Modules的权重初始化方案。这是@kaixhin对Torch7的流行nninit的一个端口。
- faster rcnn:这是更快的RCNN的PyTorch实现。该项目主要基于py-faster-rcnn和TFFRCNN。有关R-CNN的详细信息,请参阅文章更快的R-CNN:通过区域提案网络实现实时目标检测由邵少卿,何开明,Ross Girshick,孙健
- doomnet:PyTorch的Doom-net版本在ViZDoom环境中实现了一些RL模型。
- flownet:Dosovitskiy等人的Pytorch实施FlowNet。
- sqeezenet:在pytorch中实现Squeezenet,在CIFAR10数据上实现####预训练模型计划在cifar 10上训练模型并添加块连接。
- WassersteinGAN:在pytorch中的wassersteinGAN。
- optnet:此存储库由Brandon Amos和J. Zico Kolter提供,并包含PyTorch源代码,用于在我们的论文OptNet中重现实验:可区分优化作为神经网络中的一层。
- qp solver:PyTorch的快速且可微分的QP求解器。由Brandon Amos和J. Zico Kolter精心打造。
- Continuous Deep Q-Learning with Model-based Acceleration:基于模型的加速重新实现连续深度Q学习。
- Learning to learn by gradient descent by gradient descent:PyTorch实现学习通过梯度下降的梯度下降来学习。
- fast-neural-style:快速神经风格的 pytorch实现,该模型使用实时样式转移和超分辨率的感知损失中描述的方法以及实例规范化。
- PytorchNeuralStyleTransfer:在Pytorch中实现神经样式传递。
- Fast Neural Style for Image Style Transform by Pytorch:Pytorch图像风格变换的快速神经风格。
- neural style transfer:通过Leon A. Gatys,Alexander S. Ecker和Matthias Bethge开发的神经风格算法(https://arxiv.org/abs/1508.06576)介绍PyTorch。
- VIN_PyTorch_Visdom:价值迭代网络(VIN)的PyTorch实现:清洁,简单和模块化。Visdom中的可视化。
- YOLO2:PyTorch中的YOLOv2。
- attention-transfer:pytorch中的注意力转移,请阅读此处的论文。
- SVHNClassifier:使用深度卷积神经网络从街景图像中进行多位数识别的 PyTorch实现。
- pytorch-deform-conv:Deformable Convolution的PyTorch实现。
- 开始-pytorch:PyTorch执行的开始:边界平衡点创成对抗性的网络。
- treelstm.pytorch:PyTorch中的树LSTM实现。
- AGE:Dmitry Ulyanov,Andrea Vedaldi和Victor Lempitsky的论文“Adversarial Generator-Encoder Networks”的代码,可以在这里找到
- ResNeXt.pytorch:使用pytorch重现ResNet-V3(深度神经网络的聚合残差变换)。
- pytorch-rl:使用pytorch和visdom深度强化学习
- Deep-Leafsnap:与传统的计算机视觉方法相比,使用深度神经网络复制LeafSnap来测试准确性。
- pytorch-CycleGAN-and-pix2pix:用于未配对和成对图像到图像转换的PyTorch实现。
- A3C-PyTorch:PyTorch在PyTorch中实现Advantage异步演员评论算法(A3C)
- pytorch-value-iteration-networks:Pytorch实现价值迭代网络(NIPS 2016年最佳论文)
- PyTorch-Style-Transfer:用于实时传输的多样式生成网络的PyTorch实现
- pytorch-deeplab-resnet:pytorch-deeplab-resnet-model。
- pointnet.pytorch:“PointNet:用于3D分类和分割的点集的深度学习”的pytorch实现https://arxiv.org/abs/1612.00593
- pytorch-playground:pytorch中的基础预训练模型和数据集(MNIST,SVHN,CIFAR10,CIFAR100,STL10,AlexNet,VGG16,VGG19,ResNet,Inception,SqueezeNet)。
- pytorch-dnc:神经图灵机(NTM)和可分辨神经计算机(DNC)与pytorch和visdom。
- pytorch_image_classifier:使用Pytorch的最小但实用的图像分类器Pipline,ResNet18上的Finetune,在自己的小数据集上获得99%的准确度。
- mnist-svhn-transfer:用于域转移(最小)的PyTorch实现CycleGAN和SGAN。
- pytorch-yolo2:pytorch-yolo2
- dni:在Pytorch中使用合成梯度实现解耦神经接口
- wgan-gp:文章“改进Wasserstein GAN训练”的实施。
- pytorch-seq2seq-intent-parsing:使用seq2seq +注意在PyTorch中进行Intent解析和插槽填充
- pyTorch_NCE:pyTorch的噪声对比度估计算法的实现。工作,但效率不高。
- molencoder:PyTorch中的Molecular AutoEncoder
- GAN-weight-norm:“关于生成对抗网络中批量和权重标准化的影响”的代码
- lgamma:PyTorch的polygamma,lgamma和beta函数的实现
- bigBatch:用于生成结果的代码“更长时间训练,更好地概括:缩小神经网络大批量训练中的泛化差距”
- rl_a3c_pytorch:为Atari 2600实施A3C LSTM的强化学习。
- pytorch-retraining:为PyTorch的模型动物园(torchvision)转移学习枪战
- nmp_qc:计算机视觉的神经消息传递
- grad-cam:Grator-CAM的Pytorch实现
- pytorch-trpo:PyTorch实施信任区域政策优化(TRPO)
- pytorch-explain-black-box:PyTorch通过有意义的扰动实现黑盒子的可解释解释
- vae_vpflows:PyTorch中用于凸组合线性IAF和Householder Flow的代码,JM Tomczak和M. Welling https://jmtomczak.github.io/deebmed.html
- relational-networks:Pytorch实现的“关系推理的简单神经网络模块”(关系网络)https://arxiv.org/pdf/1706.01427.pdf
- vqa.pytorch:Pytorch中的视觉问题回答
- end-to-end-negotiator:交易还是不交易?谈判对话的端到端学习
- odin-pytorch:神经网络中分布式实例的原理检测。
- FreezeOut:逐步冻结层加速神经网络训练。
- ARAE:Zhao,Kim,Zhang,Rush和LeCun 撰写的“用于生成离散结构的异常正则化自动编码器”的代码。
- forward-thinking-pytorch:Pytorch实施的“前瞻思维:一次构建和训练神经网络一层” https://arxiv.org/pdf/1706.02480.pdf
- context_encoder_pytorch:PyTorch实现Context编码器
- attention-is-all-you-need-pytorch:变形金刚模型的PyTorch实现在“注意就是你所需要的一切”中。https://github.com/thnkim/OpenFacePytorch
- OpenFacePytorch:使用OpenFace的nn4.small2.v1.t7模型的PyTorch模块
- neural-combinatorial-rl-pytorch:具有强化学习的神经组合优化的PyTorch实现。95. pytorch-nec:神经情景控制(NEC)的PyTorch实现
- seq2seq.pytorch:使用PyTorch进行序列到序列学习
- Pytorch-Sketch-RNN:arxiv.org/abs/1704.03477的pytorch实现
- pytorch-pruning:PyTorch实现[1611.06440]修剪卷积神经网络进行资源有效推理
- DrQA:阅读维基百科以解答开放域问题的pytorch实现。
- YellowFin_Pytorch:自动调整动量SGD优化器
- samplernn-pytorch:SampleRNN的PyTorch实现:无条件端到端神经音频生成模型。
- AEGeAN:具有AE稳定性的更深的DCGAN
- / pytorch-SRResNet:使用生成对抗网络实现照片真实单图像超分辨率的pytorch实现方法:1609.04802v2
- vsepp:论文的代码“VSE ++:改进的视觉语义嵌入”
- Pytorch-DPPO:分布式近端策略优化的Pytorch实现:arxiv.org/abs/1707.02286
- UNIT:PyTorch实现我们的耦合VAE-GAN算法用于无监督的图像到图像转换
- efficient_densenet_pytorch:DenseNets的内存高效实现
- tsn-pytorch:PyTorch中的时间段网络(TSN)。
- SMASH:一种有效探索神经架构的实验技术。
- pytorch-retinanet:PyTorch中的RetinaNet
- biogans:支持ICCV 2017论文“生物图像合成的GAN”的实施。
- Semantic Image Synthesis via Adversarial Learning:在ICCV 2017中的文章“通过对抗性学习进行语义图像合成”的PyTorch实现。
- fmpytorch:cython中的分解机器模块的PyTorch实现。
- ORN:2017年CVPR中“定向响应网络”一文的PyTorch实现。
- pytorch-maml:MAML的PyTorch实现:arxiv.org/abs/1703.03400
- pytorch-generative-model-collections:Pytorch版本中生成模型的集合。
- vqa-winner-cvprw-2017:Pytorch在CVPR'17中实施了VQA Chllange研讨会的获奖者。
- tacotron_pytorch:PyTorch实现了Tacotron语音合成模型。
- pspnet-pytorch:PSPNet分段网络的PyTorch实现
- LM-LSTM-CRF:使用任务感知语言模型增强序列标记http://arxiv.org/abs/1709.04109
- face-alignment:Pytorch实施论文“我们在多大程度上解决了2D和3D人脸对齐问题?(以及230,000个3D面部标志的数据集)”,ICCV 2017
- DepthNet:关于Still Box数据集的PyTorch DepthNet培训。
- EDSR-PyTorch:PyTorch版本的文章'用于单图像超分辨率的增强型深度残留网络'(CVPRW 2017)
- e2c-pytorch:在PyTorch中嵌入Control实现。
- 3D-ResNets-PyTorch:用于动作识别的3D ResNets。
- bandit-nmt:这是我们的EMNLP 2017论文“使用模拟人体反馈进行强盗神经机器翻译的强化学习”的代码回购,它在神经编码器 - 解码器模型之上实现了A2C算法,并在模拟噪声奖励下对组合进行基准测试。
- pytorch-a2c-ppo-acktr:使用Kronecker因子近似(ACKTR)进行深度强化学习的Advantage Actor Critic(A2C),近端策略优化(PPO)和可扩展信任域方法的PyTorch实现。
- zalando-pytorch:来自Zalando 的Fashion-MNIST数据集的各种实验 。
- sphereface_pytorch:SphereFace的PyTorch实现。
- Categorical DQN:从分布视角看强化学习的分类DQN的PyTorch实现。
- pytorch-ntm:pytorch ntm实现。
- mask_rcnn_pytorch:在PyTorch中屏蔽RCNN。
- graph_convnets_pytorch:图形ConvNets,NIPS'16的PyTorch实现
- pytorch-faster-rcnn:基于Xinlei Chen的tf-faster-rcnn的更快的RCNN检测框架的pytorch实现。
- torchMoji:DeepMoji模型的pyTorch实现:用于分析情绪,情绪,讽刺等的最先进的深度学习模型。
- semantic-segmentation-pytorch:用于MIT ADE20K数据集的语义分割/场景解析的Pytorch实现
- pytorch-qrnn:准回归神经网络的PyTorch实现 - 比NVIDIA的cuDNN LSTM快16倍
- pytorch-sgns:PyTorch中的Skipgram负抽样。
- SfmLearner-Pytorch:来自Tinghui Zhou等人的Pytorch版本的SfmLearner。
- deformable-convolution-pytorch:可变形卷积的PyTorch实现。
- skip-gram-pytorch:一个完整的pytorch实现的skipgram模型(带子采样和负采样)。使用Spearman的秩相关性来测试嵌入结果。
- stackGAN-v2:用于再现StackGAN_v2的Pytorch实现结果StackGAN ++:堆叠生成对抗网络的逼真图像合成由Han Zhang *,Tao Xu *,Hongsheng Li,Shaoting Zhang,Xiaoangang Wang,Xiaolei Huang,Dimitris Metaxas。
- self-critical.pytorch:用于图像字幕的自我关键序列训练的非官方pytorch实现。
- pygcn:PyTorch中的图形卷积网络。
- dnc:可分辨的神经计算机,用于Pytorch
- prog_gans_pytorch_inference:用CelebA快照推断“GAN渐进式增长”的PyTorch。
- pytorch-capsule:Pytorch实现Hinton的胶囊之间的动态路由。
- PyramidNet-PyTorch:PyramidNets的PyTorch实现(Deep Pyramidal Residual Networks,arxiv.org / abs / 1610.02915)
- radio-transformer-networks:来自“物理层深度学习简介”论文的无线电变压器网络的PyTorch实现。arxiv.org/abs/1702.00832
- honk:为关键词识别谷歌的TensorFlow细胞神经网络的PyTorch重新实现。
- DeepCORAL:PyTorch实现的“Deep CORAL:Deep Domain Adaptation的相关对齐”,ECCV 2016
- pytorch-pose:用于2D人体姿势估计的PyTorch工具包。
- lang-emerge-parlai:使用PyTorch和ParlAI实施EMNLP 2017论文“自然语言不会在Multi-Agent对话中自然出现”
- Rainbow:彩虹:结合深层强化学习的改进
- pytorch_compact_bilinear_pooling v1:此存储库具有Compact Bilinear Pooling的纯Python实现和PyTorch的Count Sketch。
- CompactBilinearPooling-Pytorch v2 :( Yang Gao,et al。)Compact Cilinear Pooling的Pytorch实现。
- FewShotLearning:Pytorch实施的论文“优化作为少数镜头学习的模型”
- meProp:“meProp的代码:用于加速深度学习和减少过度拟合的Sparsified Back传播”。
- SFD_pytorch:单镜头不变脸部检测器的PyTorch实现。
- GradientEpisodicMemory:使用GEM进行连续学习:梯度情景记忆。https://arxiv.org/abs/1706.08840
- DeblurGAN:Pytorch实施的论文DeblurGAN:使用条件对抗网络的盲运动去模糊。
- StarGAN:StarGAN:用于多域图像到图像转换的统一生成对抗网络。
- CapsNet-pytorch:NIPS 2017纸张胶囊间动态路由的PyTorch实现。
- CondenseNet:CondenseNet:使用学习集团卷积的高效密集网络。
- deep-image-prior:使用神经网络进行图像恢复但没有学习。
- deep-head-pose:使用PyTorch进行深度学习头部姿势估计。
- Random-Erasing:此代码具有“随机擦除数据扩充”一文的源代码。
- FaderNetworks:推子网络:通过滑动属性操纵图像 - NIPS 2017
- FlowNet 2.0:FlowNet 2.0:使用Deep Networks进行光流估计的演进
- pix2pixHD:使用条件GAN合成和操作2048x1024图像tcwang0509.github.io/pix2pixHD
- pytorch-smoothgrad:PyTorch中的SmoothGrad实现
- RetinaNet:PyTorch中RetinaNet的实现。
- faster-rcnn.pytorch:这个项目是一个更快的R-CNN实现,旨在加速对更快的R-CNN对象检测模型的训练。
- mixup_pytorch:PyTorch实现的文章混合:超越PyTorch中的经验风险最小化。
- inplace_abn:用于内存优化的DNN培训的就地激活BatchNorm
- pytorch-pose-hg-3d:用于3D人体姿势估计的PyTorch实现
- nmn-pytorch:Pytorch中VQA的神经模块网络。
- bytenet:来自“线性时间神经机器翻译”论文的字节网的 Pytorch实现
- bottom-up-attention-vqa:vqa,自下而上注意力,pytorch
- yolo2-pytorch:YOLOv2是最受欢迎的单级物体探测器之一。该项目采用PyTorch作为提高生产力的开发框架,并利用ONNX将模型转换为Caffe 2到有利于工程的部署。
- reseg-pytorch:ReSeg的PyTorch实现(arxiv.org/pdf/1511.07053.pdf)
- binary-stochastic-neurons:PyTorch中的二元随机神经元。
- pytorch-pose-estimation:实时多人姿态估计项目的PyTorch实现。
- interaction_network_pytorch:用于学习对象,关系和物理的交互网络的Pytorch实现。
- NoisyNaturalGradient:Pytorch实施论文“嘈杂的自然梯度作为变分推理”。
- ewc.pytorch:James Kirkpatrick等人提出的弹性重量合并(EWC)的实现。克服2016年神经网络中的灾难性遗忘(10.1073 / pnas.1611835114)。
- pytorch-zssr:使用深度内部学习的1712.06087“零射击”超分辨率的PyTorch实现
- deep_image_prior:PyTorch中 Deep Image Prior(Ulyanov等,2017)的图像重建方法的实现。
- pytorch-transformer:pytorch实现注意力就是你所需要的。
- DeepRL-Grounding:这是针对任务导向语言接地的AAAI-18纸张门控注意结构的PyTorch实现
- deep-forecast-pytorch:使用PyTorch中的LSTM进行风速预测(arxiv.org/pdf/1707.08110.pdf)
- cat-net:Canonical Appearance Transformations
- minimal_glo:从“优化生成网络的潜在空间”一文中生成潜在优化的最小PyTorch实现
- LearningToCompare-Pytorch:Pytorch Paper for Paper:学习比较:少数人学习的关系网络。
- poincare-embeddings:PyTorch实施NIPS-17论文“用于学习分层表示的Poincaré嵌入”。
- pytorch-trpo(Hessian-vector product version):这是PyTorch实现的“信任区域策略优化(TRPO)”,具有精确的Hessian向量积而不是有限差分近似。
- ggnn.pytorch:门控图序列神经网络(GGNN)的PyTorch实现。
- visual-interaction-networks-pytorch:这是使用pytorch进行深度视觉交互网络的一种实现
- adversarial-patch:PyTorch实现的对抗补丁。
- Prototypical-Networks-for-Few-shot-Learning-PyTorch:在Pytorch中实现少数镜头学习的原型网络(arxiv.org/abs/1703.05175)
- Visual-Feature-Attribution-Using-Wasserstein-GANs-Pytorch:在PyTorch中使用Wasserstein GAN(arxiv.org/abs/1711.08998)实现视觉特征归因。
- PhotographicImageSynthesiswithCascadedRefinementNetworks-Pytorch:具有级联细化网络的摄影图像合成 - Pytorch实现
- ENAS-pytorch:PyTorch实现“通过参数共享进行高效的神经架构搜索”。
- Neural-IMage-Assessment:神经IMage 评估的PyTorch实现。
- proxprop:近端反向传播 - 一种神经网络训练算法,采用隐式梯度步骤而不是显式梯度步骤。
- FastPhotoStyle:照片般逼真的图像样式化的封闭形式解决方案
- Deep-Image-Analogy-PyTorch:基于pytorch的Deep-Image-Analogy的python实现。
- Person-reID_pytorch:PyTorch for Person re-ID。
- pt-dilate-rnn:pytorch中的扩张RNN。
- pytorch-i-revnet:i-RevNets的 Pytorch实现。
- OrthNet:用于生成正交多项式的TensorFlow和PyTorch图层。
- DRRN-pytorch:超分辨率深度递归残差网络(DRRN)的实现,CVPR 2017
- shampoo.pytorch:洗发水的实施。
- Neural-IMage-Assessment 2:PyTorch实现神经IMage评估。
- TCN:序列建模基准和时间卷积网络locuslab / TCN
- DCC:此存储库包含用于再现Deep Continuous Clustering纸张结果的源代码和数据。
- packnet:PackNet代码:通过迭代修剪将多个任务添加到单个网络arxiv.org/abs/1711.05769
- PyTorch-progressive_growing_of_gans:PyTorch实现GAN的逐步增长,以提高质量,稳定性和变异性。
- nonauto-nmt:PyTorch实现“非自回归神经机器翻译”
- PyTorch-GAN:生成对抗网络的PyTorch实现。
- PyTorchWavelets:在Torrence和Compo(1998)中发现的小波分析的PyTorch实现
- pytorch-made:PyTorch中的 MADE(Masked Autoencoder Density Estimation)实现
- VRNN:来自序列数据的循环潜变量模型的变分RNN(VRNN)的Pytorch实现。
- flow:Pytorch实施ICLR 2018论文物理过程的深度学习:整合先前的科学知识。
- deepvoice3_pytorch:PyTorch实现基于卷积网络的文本到语音合成模型
- psmm:Pointer Sentinel混合模型的实现,如Stephen Merity等人的论文所述。
- tacotron2:Tacotron 2 - PyTorch实现,具有比实时推断更快的速度。
- AccSGD:实现Accelerated SGD算法的pytorch代码。
- QANet-pytorch:使用PyTorch实现QANet(EM / F1 = 70.5 / 77.2,在一个1080Ti卡上20个epoches后约20个小时。)
- ConvE:卷积2D知识图嵌入
- Structured-Self-Attention:本文的实施结构性自我意识句子嵌入,发布于2017年ICLR:arxiv.org/abs/1703.03130。
- graphage-simple:GraphSAGE的简单参考实现。
- Detectron.pytorch:Detectron的pytorch实现。既可以从头开始训练,也可以直接从预训练的Detectron重量推断。
- R2Plus1D-PyTorch:基于R2Plus1D基于卷积的ResNet架构的PyTorch实现,该文件在“动态识别的时空卷积近观”中有所描述。
- StackNN:用于神经网络的可区分堆栈的PyTorch实现。
- translagent:多Agent通信中的紧急翻译代码。
- ban-vqa:用于视觉问答的双线性注意网络。
- pytorch-openai-transformer-lm:这是TensorFlow代码的PyTorch实现,由OpenAI的论文“生成预训练提高语言理解”提供,由Alec Radford,Karthik Narasimhan,Tim Salimans和Ilya Sutskever提供。
- T2F:使用深度学习进行文本到面的生成。该项目结合了最近的两个架构StackGAN和ProGAN,用于从文本描述中合成面部。
- pytorch - fid:PyTorch的Fréchet初始距离(FID得分)
- vae_vpflows:PyTorch中用于凸组合线性IAF和Householder Flow的代码,JM Tomczak和M. Welling jmtomczak.github.io/deebmed.html
- CoordConv-pytorch:CoordConv的Pytorch实现在'卷入神经网络的有趣失败和CoordConv解决方案'论文中引入。(arxiv.org/pdf/1807.03247.pdf)
- SDPoint:在CVPR 2018中发布的“卷积网络中用于成本可调推理和改进正则化的随机下采样”的实施。
- SRDenseNet-pytorch:SRDenseNet-pytorch(ICCV_2017)
- GAN_stability:论文代码“GAN实际上哪些训练方法能够融合?(ICML 2018)”
- Mask-RCNN:Mask RCNN架构的PyTorch实现,作为使用PyTorch的介绍
- pytorch-coviar:压缩视频动作识别
- PNASNet.pytorch:ImageNet上PNASNet-5的PyTorch实现。
- NALU-pytorch:来自神经算术逻辑单元的NAC / NALU的基本pytorch实现arxiv.org/pdf/1808.00508.pdf
- LOLA_DiCE:Pytorch使用DiCE实现LOLA(arxiv.org/abs/1709.04326)(arxiv.org/abs/1802.05098)
- generative-query-network-pytorch:PyTorch中的生成查询网络(GQN),如“神经场景表示和渲染”中所述
- pytorch_hmax:在PyTorch中实现HMAX视觉模型。
- FCN-pytorch-easyiest:尝试成为FCN(完全控制网络)中最简单,最容易使用的pytorch实现
- transducer:使用PyTorch绑定实现快速序列传感器。
- AVO-pytorch:PyTorch中对抗变分优化的实现。
- HCN-pytorch:{for-occurrence特征学习从骨架数据学习用于动作识别和分层聚合检测}的pytorch重新实现}。
- binary-wide-resnet:McDonnel的具有1位权重的宽剩余网络的PyTorch实现(ICLR 2018)
- piggyback:Code for Piggyback:通过学习掩盖权重使单个网络适应多个任务arxiv.org/abs/1801.06519
- vid2vid:Pytorch实现我们的高分辨率(例如2048x1024)逼真的视频到视频转换方法。
- poisson-convolution-sum:实现一个无限的泊松加权卷积和
- tbd-nets:PyTorch实现“透明设计:缩小视觉推理中性能和可解释性之间的差距”arxiv.org/abs/1803.05268
- attn2d:普遍关注:用于序列到序列预测的2D卷积网络
- yolov3:YOLOv3:PyTorch的培训和推理pjreddie.com/darknet/yolo
- deep-dream-in-pytorch:Pytorch实现的DeepDream计算机视觉算法。
- pytorch-flows:用于密度估计的算法的PyTorch实现
- quantile-regression-dqn-pytorch:分位数回归DQN是最小工作实例
- relational-rnn-pytorch:PyTorch中DeepMind关系递归神经网络的一种实现。
- DEXTR-PyTorch:Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr
- PyTorch_GBW_LM:Google Billion Word Dataset的PyTorch语言模型。
- Pytorch-NCE:用Pytorch编写的softmax输出的噪声对比度估计
- 生成模型:注释,可理解和可视化解释的PyTorch实现:VAE,BIRVAE,NSGAN,MMGAN,WGAN,WGANGP,LSGAN,DRAGAN,BEGAN,RaGAN,InfoGAN,fGAN,FisherGAN。
- convnet-aig:具有自适应推理图的卷积网络的PyTorch实现。
- integrated-gradient-pytorch:这是本文的pytorch实现 - Deep Networks的Axiomatic Attribution。
- MalConv-Pytorch:Maltorv的Pytorch实现。275,trellisnet:用于序列建模的Trellis Networks
Pytorch其他
- the-incredible-pytorch:The Incredible PyTorch:一系列有关PyTorch的教程,论文,项目,社区等的精选列表。
- generative models:生成模型的集合,例如Tensorflow,Keras和Pytorch中的GAN,VAE。http://wiseodd.github.io
- pytorch vs tensorflow:reddit上的一个信息性线程。
- Pytorch讨论论坛
- pytorch notebook:docker-stack:类似于Jupyter Notebook Scientific Python Stack的项目
- drawlikebobross:使用神经网络(使用PyTorch)的力量像鲍勃罗斯一样画画!
- pytorch-tvmisc:完全用于Pytorch的多功能杂项
- pytorch-a3c-mujoco:为Mujoco健身房环境实施A3C。
- PyTorch在5分钟内完成。
- pytorch_chatbot:使用PyTorch实现的奇妙ChatBot。
- malmo-challenge:Malmo Collaborative AI Challenge - Team Pig Catcher
- sketchnet:一种模型,它采用图像并生成处理源代码以重新生成该图像
- Deep-Learning-Boot-Camp:一个非营利性社区运行,为期5天的深度学习训练营http://deep-ml.com。
- Amazon_Forest_Computer_Vision:使用PyTorch / Keras和许多PyTorch技巧的卫星图像标记代码。讨价还价的比赛。
- AlphaZero_Gomoku:Gomoku的AlphaZero算法的实现(也称为Gobang或连续五个)
- pytorch-cv:物体检测,分割和姿态估计的回购。
- deep-person- reid:Pytorch实施深层人员重新识别方法。
- pytorch-template:PyTorch模板项目
- 使用Pytorch TextBook进行深度学习使用PyTorch在文本和视觉中构建神经网络模型的实用指南。在亚马逊 github代码回购购买
- compare-tensorflow-pytorch:比较Tensorflow中编写的层与Pytorch中编写的层之间的输出。
- hasktorch:Haskell中的张量和神经网络
- Pytorch深度学习使用PyTorch进行深度学习教你如何使用Python和PyTorch实现深度学习算法。
- nimtorch:PyTorch - Python + Nim
- derplearning:自驾车辆代码。
反馈:如果您有任何想法或者您希望将任何其他内容添加到此列表中,请随时提供。