segmentation_models.pytorch

segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-

Python9718mit

5 days ago

computer-visiondeeplab-v3-plusdeeplabv3

pytorch-grad-cam

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transf

Python10619mit

last month

class-activation-mapscomputer-visiondeep-learning

pytorch-image-models

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, e

Python32336apache-2.0

8 hours ago

augmixconvnextdistributed-training

pytorch-fid

Compute FID scores with PyTorch.

Python3420apache-2.0

5 months ago

deep-learningfidfid-score

pytorch-frame

pytorch-frame

Tabular Deep Learning Library for PyTorch

Python547mit

5 days ago

data-framedeep-learningpytorch

tnt

A lightweight library for PyTorch training tools and utilities

Python1666other

yesterday

deep-learningmachine-learningneural-network

deep-high-resolution-net.pytorch

deep-high-resolution-net.pytorch

The project is an official implementation of our CVPR2019 paper "Deep High-Resol

Cuda4329mit

3 months ago

coco-keypoints-detectiondeep-high-resolution-netdeep-learning

pytorch-lightning

pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero cod

Python28442apache-2.0

16 hours ago

aiartificial-intelligencedata-science

lucent

lucent

Lucid library adapted for PyTorch

Python609apache-2.0

6 months ago

tch-rs

Rust bindings for the C++ api of PyTorch.

Rust4063apache-2.0

4 months ago

deep-learningmachine-learningneural-network

kaolin

kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

Python4507apache-2.0

3 days ago

3d-deep-learningartificial-intelligencecamera-api

pytorch_geometric

pytorch_geometric

Graph Neural Network Library for PyTorch

Python21431mit

5 hours ago

deep-learninggeometric-deep-learninggraph-convolutional-networks

pytracking

pytracking

Visual tracking library based on PyTorch.

Python3170gpl-3.0

4 months ago

computer-visionmachine-learningtracking

machine-learning-book

machine-learning-book

Code Repository for Machine Learning with PyTorch and Scikit-Learn

Jupyter Notebook3427mit

4 months ago

deep-learningmachine-learningneural-networks

bindsnet

bindsnet

Simulation of spiking neural networks (SNNs) using PyTorch.

Python1509agpl-3.0

last month

dynamicgpu-computingmachine-learning

skorch

A scikit-learn compatible neural network library that wraps PyTorch

Jupyter Notebook5887bsd-3-clause

18 days ago

hacktoberfesthuggingfacemachine-learning

detecto

detecto

Build fully-functioning computer vision models with PyTorch

Python613mit

4 months ago

computer-visionfaster-rcnnmachine-learning

edsnlp

Modular, fast NLP framework, compatible with Pytorch and spaCy, offering tailore

Python115bsd-3-clause

3 days ago

clinical-data-warehousedeep-learningfast

einops

Flexible and powerful tensor operations for readable and reliable code (for pyto

Python8535mit

4 days ago

chainercupydeep-learning

poutyne

poutyne

A simplified framework and utilities for PyTorch

Python569lgpl-3.0

5 months ago

data-sciencedeep-learningkeras

deepdetect

deepdetect

Deep Learning API and Server in C++14 support for PyTorch,TensorRT, Dlib, NCNN,

C++2519other

19 hours ago

caffedeep-learninggpu

sinabs

sinabs

A deep learning library for spiking neural networks which is based on PyTorch, f

Python81agpl-3.0

10 days ago

machine-learningpytorchsnn

pytorch_tabular

pytorch_tabular

A standard framework for modelling Deep Learning Models for tabular data

Python1391mit

6 hours ago

deep-learninghacktoberfestmachine-learning

TResNet

TResNet

Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Arch

Python471apache-2.0

5 days ago

accuracyarchitectureimagenet

ocaml-torch

ocaml-torch

OCaml bindings for PyTorch

OCaml411apache-2.0

last month

artificial-intelligencedeep-learninggpu

pyro

pyro

Deep universal probabilistic programming with Python and PyTorch

Python8582apache-2.0

20 days ago

bayesianbayesian-inferencedeep-learning

yolov3

yolov3

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Python10236agpl-3.0

4 days ago

deep-learningmachine-learningobject-detection

yolov5

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Python51021agpl-3.0

2 days ago

coremldeep-learningios

deeplake

deeplake

Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangCh

Python8188apache-2.0

7 hours ago

aicomputer-visioncv

transformers

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and

Python135330apache-2.0

9 hours ago

bertdeep-learningflax

determined

determined

Determined is an open-source machine learning platform that simplifies distribut

Go3045apache-2.0

10 hours ago

data-sciencedeep-learningdistributed-training

ros2_pytorch

ROS2 PyTorch template node - running PyTorch C++ models in ROS2

C++39apache-2.0

6 years ago

ros2_pytorch_cuda

ROS2 PyTorch template node - running PyTorch C++ models in ROS2

C++5apache-2.0

2 years ago

flownet2-pytorch

flownet2-pytorch

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

Python3101other

last year

pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Python30286mit

last year

deep-learningneural-networkspytorch

beauty-net

A simple, flexible, and extensible template for PyTorch. It's beautiful.

Python187mit

last year

cnncomputer-visiondeep-learning

DrQA

DrQA

A pytorch implementation of Reading Wikipedia to Answer Open-Domain Questions.

Python401

3 years ago

drqapytorchsquad

lightnlg

A minimalistic codebase for training NLG models from HuggingFace Transformers us

Python3apache-2.0

2 years ago

bartgenerationgpt-2

deep-learning-project-template

Pytorch Lightning code guideline for conferences

Python1240apache-2.0

last year

quickvision

quickvision

An Easy To Use PyTorch Computer Vision Library

Python51apache-2.0

last year

computer-visiondeep-learningpytorch

neural-dream

neural-dream

PyTorch implementation of DeepDream algorithm

Python132mit

3 years ago

computer-visiondeep-dreamdeepdream

NDScala

N-dimensional / multi-dimensional arrays (tensors) in Scala 3. Think NumPy ndarr

Scala47agpl-3.0

2 years ago

array-programmingdottymachine-learning

CNN_keras

CNN | Keras/PyTorch | CAPTCHA recognition(卷积神经网络、Keras/PyTorch框架、验证码识别)

Python287mit

5 years ago

captchacnn-kerasmachine-learning

TorchSeg

TorchSeg

Fast, modular reference implementation and easy training of Semantic Segmentatio

Python1403mit

5 years ago

pytorch2keras

PyTorch to Keras model convertor

Python858mit

2 years ago

converterdartsdeep-learning

SFA3D

SFA3D

Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The

Python1034mit

last year

3d-object-detectionbevmapcenter

torch-points3d

torch-points3d

Pytorch framework for doing deep learning on point clouds.

220other

3 years ago

yolo-v5-tflite-model

YOLOv5 - most advanced vision AI model for object detection. Natively implemente

Python17apache-2.0

3 years ago

neural-style-pt

neural-style-pt

PyTorch implementation of neural style transfer algorithm

Python840mit

2 years ago

deep-learningdeep-styleneural-style

SLM-Lab

SLM-Lab

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of t

Python1256mit

2 years ago

a2ca3cbenchmark

PyTorchCV

PyTorchCV

PyTorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision.

Shell47apache-2.0

6 years ago

graph-2-text

Graph to sequence implemented in Pytorch combining Graph convolutional networks

Python151mit

5 years ago

gradslam

gradslam

gradslam is an open source differentiable dense SLAM library for PyTorch

Python1321mit

last year

3d-reconstructiondeep-learningmachine-learning

GeoDA

GeoDA

This repository contains the official PyTorch implementation of GeoDA algorithm.

Python32

4 years ago