pytorch-implementation

👁️ 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...

python pytorch ssim psnr ms-ssim iqa image-quality-assessment blind-image-quality-assessment image-aesthetic-assessment pytorch-implementation

UpdatedJul 2, 2025
Python

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

UpdatedNov 17, 2022
Jupyter Notebook

sthalles /SimCLR

Star2.4k

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

UpdatedMar 4, 2024
Jupyter Notebook

sjtuytc /UnboundedNeRFPytorch

Star1.3k

State-of-the-art, simple, fast unbounded / large-scale NeRFs.

deep-learning pytorch chinese-translation nerf classified unbounded pytorch-implementation view-synthesis block-nerf weekly-nerf

UpdatedJun 11, 2024
Python

MIC-DKFZ /medicaldetectiontoolkit

Star1.3k

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 medical-imaging segmentation object-detection medical-image-computing semantic-segmentation 3d-models medical-image-processing mask-rcnn u-net medical-image-analysis pytorch-cnn retina-net 3d-object-detection pytorch-implementation pytorch-deeplearning retina-unet 3d-mask-rcnn

UpdatedJun 17, 2024
Python

dailenson /SDT

Star1.2k

This repository is the official implementation of Disentangling Writer and Character Styles for Handwriting Generation (CVPR 2023)

computer-vision deep-learning transformer gmm generative-models handwriting-generation multimodal pytorch-implementation contrastive-learning

UpdatedJun 26, 2025
Python

bentrevett /pytorch-image-classification

Star1k

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

UpdatedDec 28, 2021
Jupyter Notebook

LibCity /Bigscity-LibCity

Star1k

LibCity: An Open Library for Urban Spatial-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

UpdatedDec 18, 2024
Python

piEsposito /blitz-bayesian-deep-learning

Star967

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

UpdatedSep 25, 2023
Python

hkproj /pytorch-stable-diffusion

Star919

Stable Diffusion implemented from scratch in PyTorch

pytorch paper-implementations pytorch-implementation diffusion-models stable-diffusion latent-diffusion-models

UpdatedOct 22, 2024
Jupyter Notebook

ddlBoJack /emotion2vec

Star901

[ACL 2024] Official PyTorch code for extracting features and training downstream models with emotion2vec: Self-Supervised Pre-Training for Speech Emotion Representation

speech-emotion-recognition pytorch-implementation iemocap speech-representation