segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-
Python9718mit
5 days ago
computer-visiondeeplab-v3-plusdeeplabv3
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
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
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
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
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero cod
Python28442apache-2.0
16 hours ago
aiartificial-intelligencedata-science
tch-rs
Rust bindings for the C++ api of PyTorch.
Rust4063apache-2.0
4 months ago
deep-learningmachine-learningneural-network
kaolin
A PyTorch Library for Accelerating 3D Deep Learning Research
Python4507apache-2.0
3 days ago
3d-deep-learningartificial-intelligencecamera-api
pytorch_geometric
Graph Neural Network Library for PyTorch
Python21431mit
5 hours ago
deep-learninggeometric-deep-learninggraph-convolutional-networks
pytracking
Visual tracking library based on PyTorch.
Python3170gpl-3.0
4 months ago
computer-visionmachine-learningtracking
machine-learning-book
Code Repository for Machine Learning with PyTorch and Scikit-Learn
Jupyter Notebook3427mit
4 months ago
deep-learningmachine-learningneural-networks
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
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
A simplified framework and utilities for PyTorch
Python569lgpl-3.0
5 months ago
data-sciencedeep-learningkeras
deepdetect
Deep Learning API and Server in C++14 support for PyTorch,TensorRT, Dlib, NCNN,
C++2519other
19 hours ago
caffedeep-learninggpu
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
A standard framework for modelling Deep Learning Models for tabular data
Python1391mit
6 hours ago
deep-learninghacktoberfestmachine-learning
TResNet
Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Arch
Python471apache-2.0
5 days ago
accuracyarchitectureimagenet
ocaml-torch
OCaml bindings for PyTorch
OCaml411apache-2.0
last month
artificial-intelligencedeep-learninggpu
pyro
Deep universal probabilistic programming with Python and PyTorch
Python8582apache-2.0
20 days ago
bayesianbayesian-inferencedeep-learning
yolov3
YOLOv3 in PyTorch > ONNX > CoreML > TFLite
Python10236agpl-3.0
4 days ago
deep-learningmachine-learningobject-detection
yolov5
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
Python51021agpl-3.0
2 days ago
coremldeep-learningios
deeplake
Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangCh
Python8188apache-2.0
7 hours ago
aicomputer-visioncv
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and
Python135330apache-2.0
9 hours ago
bertdeep-learningflax
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
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
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
An Easy To Use PyTorch Computer Vision Library
Python51apache-2.0
last year
computer-visiondeep-learningpytorch
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
Fast, modular reference implementation and easy training of Semantic Segmentatio
Python1403mit
5 years ago
SFA3D
Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The
Python1034mit
last year
3d-object-detectionbevmapcenter
yolo-v5-tflite-model
YOLOv5 - most advanced vision AI model for object detection. Natively implemente
Python17apache-2.0
3 years ago
neural-style-pt
PyTorch implementation of neural style transfer algorithm
Python840mit
2 years ago
deep-learningdeep-styleneural-style
SLM-Lab
Modular Deep Reinforcement Learning framework in PyTorch. Companion library of t
Python1256mit
2 years ago
a2ca3cbenchmark
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 is an open source differentiable dense SLAM library for PyTorch
Python1321mit
last year
3d-reconstructiondeep-learningmachine-learning
GeoDA
This repository contains the official PyTorch implementation of GeoDA algorithm.
Python32
4 years ago