Pytorch Libraries - Page 8

Semantic Segmentation Architectures Implemented in PyTorch

Python 3054Updated: 4 y ago License: Permissive (MIT)

Code for robust monocular depth estimation described in "Ranftl et. al., Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer, TPAMI 2022"

Python 3028Updated: 2 y ago License: Permissive (MIT)

RepVGG: Making VGG-style ConvNets Great Again

Python 3023Updated: 2 y ago License: Permissive (MIT)

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.

Jupyter Notebook 3023Updated: 2 y ago License: Permissive (MIT)

DeepFill v1/v2 with Contextual Attention and Gated Convolution, CVPR 2018, and ICCV 2019 Oral

Python 3018Updated: 2 y ago License: Proprietary (Proprietary)

AI实战-practicalAI 中文版

Jupyter Notebook 3018Updated: 2 y ago License: Permissive (MIT)

Code for reproducing results in "Glow: Generative Flow with Invertible 1x1 Convolutions"

Python 3014Updated: 2 y ago License: Permissive (MIT)

Largest multi-label image database; ResNet-101 model; 80.73% top-1 acc on ImageNet

Python 3010Updated: 2 y ago License: Proprietary (Proprietary)

A high-performance distributed training framework for Reinforcement Learning

Python 3000Updated: 2 y ago License: Permissive (Apache-2.0)

PyTorch0.4 implementation of: actor critic / proximal policy optimization / acer / ddpg / twin dueling ddpg / soft actor critic / generative adversarial imitation learning / hindsight experience replay

Jupyter Notebook 2994Updated: 2 y ago License: No License (No License)

SOTA Re-identification Methods and Toolbox

Python 2970Updated: 2 y ago License: Permissive (Apache-2.0)

OpenMMLab Video Perception Toolbox. It supports Video Object Detection (VID), Multiple Object Tracking (MOT), Single Object Tracking (SOT), Video Instance Segmentation (VIS) with a unified framework.

Python 2968Updated: 2 y ago License: Permissive (Apache-2.0)

FedML - The federated learning and analytics library enabling secure and collaborative machine learning on decentralized data anywhere at any scale. Supporting large-scale cross-silo federated learning, cross-device federated learning on smartphones/IoTs, and research simulation. MLOps and App Marketplace are also enabled (https://open.fedml.ai).

Python 2950Updated: 2 y ago License: Permissive (Apache-2.0)

Modularized Implementation of Deep RL Algorithms in PyTorch

Python 2938Updated: 2 y ago License: Permissive (MIT)

A tensorflow implementation of EAST text detector

C++ 2937Updated: 2 y ago License: Strong Copyleft (GPL-3.0)

Rust bindings for the C++ api of PyTorch.

Rust 2930Updated: 2 y ago License: Permissive (Apache-2.0)

🏅 Collection of Kaggle Solutions and Ideas 🏅

HTML 2924Updated: 2 y ago License: Permissive (MIT)

Pytorch implementations of various Deep NLP models in cs-224n(Stanford Univ)

Jupyter Notebook 2916Updated: 2 y ago License: Permissive (MIT)

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Python 2915Updated: 2 y ago License: Proprietary (Proprietary)

Time series forecasting with PyTorch

Python 2908Updated: 2 y ago License: Permissive (MIT)

PyTorch implementation of DQN, AC, ACER, A2C, A3C, PG, DDPG, TRPO, PPO, SAC, TD3 and ....

Python 2907Updated: 2 y ago License: Permissive (MIT)

Massively Parallel Deep Reinforcement Learning. 🔥

Python 2902Updated: 2 y ago License: Proprietary (Proprietary)

An optimizer that trains as fast as Adam and as good as SGD.

Python 2898Updated: 2 y ago License: Permissive (Apache-2.0)

A model library for exploring state-of-the-art deep learning topologies and techniques for optimizing Natural Language Processing neural networks

Python 2897Updated: 2 y ago License: Permissive (Apache-2.0)

A deep learning library for video understanding research.

Python 2894Updated: 2 y ago License: Permissive (Apache-2.0)

A Simple and Versatile Framework for Object Detection and Instance Recognition

Python 2892Updated: 4 y ago License: Permissive (Apache-2.0)

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

Python 2890Updated: 3 y ago License: Permissive (Apache-2.0)

PyTorch implementation of Super SloMo by Jiang et al.

Python 2881Updated: 2 y ago License: Permissive (MIT)

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

Python 2879Updated: 2 y ago License: Proprietary (Proprietary)

Code repository for the examples from the Packt book "Learning Threejs"

JavaScript 2878Updated: 2 y ago License: No License (No License)

LPIPS metric. pip install lpips

Python 2877Updated: 2 y ago License: Permissive (BSD-2-Clause)

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch

Python 2865Updated: 2 y ago License: Permissive (Apache-2.0)

CUDA Templates for Linear Algebra Subroutines

C++ 2862Updated: 2 y ago License: Proprietary (Proprietary)

Deep recommender models using PyTorch.

Python 2858Updated: 2 y ago License: Permissive (MIT)

Siamese and triplet networks with online pair/triplet mining in PyTorch

Python 2855Updated: 2 y ago License: Permissive (BSD-3-Clause)

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Python 2846Updated: 2 y ago License: Proprietary (Proprietary)

a distributed deep learning platform

C++ 2832Updated: 2 y ago License: Permissive (Apache-2.0)

pytorch tutorial for beginners

Python 2829Updated: 2 y ago License: No License (No License)

腾讯优图高精度双分支人脸检测器

Python 2823Updated: 2 y ago License: Proprietary (Proprietary)

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

Python 2817Updated: 2 y ago License: Permissive (MIT)

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

Python 2816Updated: 2 y ago License: Permissive (Apache-2.0)

Sandbox for training deep learning networks

Python 2796Updated: 2 y ago License: Permissive (MIT)

The project is an official implement of our ECCV2018 paper "Simple Baselines for Human Pose Estimation and Tracking(https://arxiv.org/abs/1804.06208)"

Python 2786Updated: 2 y ago License: Permissive (MIT)

SSD: Single Shot MultiBox Detector | a PyTorch Tutorial to Object Detection

Python 2785Updated: 2 y ago License: Permissive (MIT)

rllab is a framework for developing and evaluating reinforcement learning algorithms, fully compatible with OpenAI Gym.

Python 2784Updated: 2 y ago License: Proprietary (Proprietary)

Visual tracking library based on PyTorch.

Python 2778Updated: 2 y ago License: Strong Copyleft (GPL-3.0)

PyTorch original implementation of Cross-lingual Language Model Pretraining.

Python 2767Updated: 2 y ago License: Proprietary (Proprietary)

Graph Attention Networks (https://arxiv.org/abs/1710.10903)

Python 2755Updated: 2 y ago License: Permissive (MIT)

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.

Python 2754Updated: 2 y ago License: Proprietary (Proprietary)

Repository for "Introduction to Artificial Neural Networks and Deep Learning: A Practical Guide with Applications in Python"

Jupyter Notebook 2751Updated: 2 y ago License: Proprietary (Proprietary)