Build software better, together

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

python deep-learning jupyter pytorch attention attention-mechanism graph-attention-networks self-attention pytorch-implementation gat graph-attention-network pytorch-gat gat-tutorial

Updated May 10, 2022
Jupyter Notebook

sthalles / SimCLR

Star

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

machine-learning deep-learning pytorch representation-learning unsupervised-learning contrastive-loss torchvision pytorch-implementation simclr

Updated Apr 20, 2022
Jupyter Notebook

MIC-DKFZ / medicaldetectiontoolkit

Star

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

deep-neural-networks deep-learning detection inference medical-imaging segmentation object-detection medical-image-computing semantic-segmentation 3d-models medical-image-processing mask-rcnn u-net medical-image-analysis 2d-3d pytorch-cnn retina-net 3d-object-detection pytorch-implementation pytorch-deeplearning retina-unet 3d-mask-rcnn

Updated Apr 4, 2022
Python

zhmiao / OpenLongTailRecognition-OLTR

Star

Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)

computer-vision deep-learning open-set long-tail pytorch-implementation cvpr2019 open-long-tail-recognition oltr

Updated Jun 15, 2020
Python

piEsposito / blitz-bayesian-deep-learning

Star

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

pytorch bayesian-neural-networks pytorch-tutorial bayesian-deep-learning pytorch-implementation bayesian-layers

Updated Jun 9, 2022
Python

bentrevett / pytorch-image-classification

Star

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.

tutorial cnn pytorch vgg lenet image-classification resnet alexnet convolutional-networks convolutional-neural-networks convolutional-neural-network pytorch-tutorial pytorch-tutorials pytorch-cnn pytorch-implmention torchvision pytorch-implementation

Updated Dec 28, 2021
Jupyter Notebook

miguelvr / dropblock

Star

Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.

machine-learning computer-vision pytorch dropout regularization convolutional-neural-networks pytorch-implementation dropblock

Updated Jul 29, 2020
Python

soobinseo / Transformer-TTS

Star

A Pytorch Implementation of "Neural Speech Synthesis with Transformer Network"

text-to-speech deep-learning pytorch tts transformer attention-mechanism pytorch-implementation attention-plots

Updated Jan 14, 2021
Python

vaibhawvipul / First-steps-towards-Deep-Learning

Star

This is an open sourced book on deep learning.

deep-neural-networks computer-vision deep-learning python3 pytorch recurrent-neural-networks artificial-intelligence how-to artificial-neural-networks convolutional-neural-networks deep-learning-tutorial pytorch-tutorial pytorch-implementation

Updated Mar 12, 2022
CSS

omerbsezer / Fast-Pytorch

Star

Pytorch Tutorial, Pytorch with Google Colab, Pytorch Implementations: CNN, RNN, DCGAN, Transfer Learning, Chatbot, Pytorch Sample Codes

python machine-learning computer-vision deep-learning cnn pytorch rnn mlp transfer-learning pytorch-tutorial rnn-pytorch colaboratory colab-notebook cnn-pytorch pytorch-implementation colab-tutorial

Updated May 7, 2019
Jupyter Notebook

d-li14 / mobilenetv3.pytorch

Star

74.3% MobileNetV3-Large and 67.2% MobileNetV3-Small model on ImageNet

imagenet pretrained-models pytorch-implementation mobilenetv3

Updated Feb 14, 2021
Python

eagle705 / pytorch-bert-crf-ner

Star

KoBERT와 CRF로 만든 한국어 개체명인식기 (BERT+CRF based Named Entity Recognition model for Korean)

nlp natural-language-processing crf pytorch named-entity-recognition korean ner bert korean-nlp attention-visualization pytorch-implementation bert-bilstm-crf huggingface bert-crf kobert kobert-crf bert-bigru-crf

Updated May 26, 2022
Jupyter Notebook

Urinx / alphafold_pytorch

Star

An implementation of the DeepMind's AlphaFold based on PyTorch for research

deep-learning contact-prediction structure-prediction protein-structure-prediction pytorch-implementation alphafold casp13

Updated Jan 23, 2021
Python

LibCity / Bigscity-LibCity

Star

LibCity: An Open Library for Urban Spatio-temporal Data Mining

deep-learning toolkit traffic eta map-matching representation-learning on-demand-service spatio-temporal traffic-prediction trajectory-prediction time-series-prediction spatio-temporal-prediction traffic-flow-prediction pytorch-implementation od-matrix traffic-forecasting estimated-time-of-arrival traffic-accident-prediction traffic-speed-prediction libcity

Updated Jun 7, 2022
Python

zyddnys / manga-image-translator

Star

Translate manga/image 一键翻译各类图片内文字 https://touhou.ai/imgtrans/

ocr deep-learning neural-network anime machine-translation manga image-processing transformer chinese-translation text-detection auto-translation inpainting text-detection-recognition pytorch-implementation japanese-translations

Updated May 20, 2022
Python

d-li14 / efficientnetv2.pytorch

Star

PyTorch implementation of EfficientNetV2 family

imagenet pytorch-implementation efficientnet efficientnetv2 icml2021

Updated Feb 20, 2022
Python

tkat0 / PyTorch_BlazeFace

Star

Unofficial PyTorch implementation of BlazeFace

pytorch-implementation blazeface

Updated Aug 9, 2021
Python

layumi / Seg-Uncertainty

Star

IJCAI2020 & IJCV 2021 🌇 Unsupervised Scene Adaptation with Memory Regularization in vivo

pytorch self-driving-car gta5 transfer-learning semantic-segmentation ijcai robotcar cityscapes domain-adaptation synthia domainadaptation pytorch-implementation mrnet ijcai2020

Updated Dec 23, 2021
Python

mehtadushy / SelecSLS-Pytorch

Star

Reference ImageNet implementation of SelecSLS CNN architecture proposed in the SIGGRAPH 2020 paper "XNect: Real-time Multi-Person 3D Motion Capture with a Single RGB Camera". The repository also includes code for pruning the model based on implicit sparsity emerging from adaptive gradient descent methods, as detailed in the CVPR 2019 paper "On implicit filter level sparsity in Convolutional Neural Networks".

sparsity deep-learning efficient cnn pytorch imagenet pruning siggraph pytorch-implementation cvpr2019 efficient-architectures

Updated Jul 23, 2020
Python

MIC-DKFZ / nnDetection

Star

nnDetection is a self-configuring framework for 3D (volumetric) medical object detection which can be applied to new data sets without manual intervention. It includes guides for 12 data sets that were used to develop and evaluate the performance of the proposed method.

detection medical medical-imaging medical-image-computing 3d-object-detection pytorch-implementation retina-unet