Pytorch Implementation Python Packages

pyiqa

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

zetascale

Build high-performance AI models with modular building blocks

sixdrepnet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

dropblock

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

chuchichaestli

Where you find all the state-of-the-art cooking utensils (salt, pepper, gradient descent... the usual).

buctoolkit

Batch-upscaled Catalysis Toolkit

deep-fake-detection

This project Deepfakes can distort our perception of the truth and we need to develop a strategy to improve their detection. Deep Fakes are increasingly detrimental to privacy, social security, and democracy. We plan to achieve better accuracy in predicting real and fake videos.. https://pypi.org/project/Deep-Fake-Detection/

adfwi-torch

An Automatic Differentiation-based Waveform Inversion Framework Implemented in PyTorch.

dgmr

Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954

blitz-bayesian-pytorch

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

fewsoldataloader

This repo hosts the PyTorch dataloader for FewSOL dataset | ICRA 2023

alicia

A CLI to download, create, modify, train, test, predict and compare an image classifiers.

modtorch

MODTORCH is a powerful meta-language for building PyTorch networks on the fly, without having to write custom PyTorch classes manually.

torch-metrics

Metrics for model evaluation in pytorch

ncrnabert

ncRNABert: Deciphering the landscape of non-coding RNA using language model

ghonn-models-pytorch

Gated Higher Order Neural Networks models. PyTorch.

explainable-cnn

📦 PyTorch based visualization package for generating layer-wise explanations for CNNs.

model-suite

Model Suite

torchsom

torchsom: The Reference PyTorch Library for Self-Organizing Maps

rllite

rl lite repository

causaltorch

CausalTorch is a PyTorch library for building generative models with explicit causal constraints. It integrates graph-based causal reasoning with deep learning to create AI systems that respect logical causal relationships.

dynenv

Dynamic Simulation Environments for Reinforcement Learning

skillful-nowcasting

PyTorch Skillful Nowcasting GAN Implementation

hift

memory-efficient fine-tuning; support 24G GPU memory fine-tuning 7B