Tensors and Dynamic neural networks in Python with strong GPU acceleration

Inference code for Llama models

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

Open Machine Learning Course

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

A resource for learning about Machine learning & Deep Learning

PyTorch implementations of deep reinforcement learning algorithms and environments

[CVPR2020] Adversarial Latent Autoencoders

pytorch structural similarity (SSIM) loss

Translate images to unseen domains in the test time with few example images.

Code for CVPR 2019 paper. BASNet: Boundary-Aware Salient Object Detection

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)

An easy to follow course of OpenCV using Python for beginners.

Self-contained, minimalistic implementation of diffusion models with Pytorch.

Here is a pytorch implementation of deeplabv3+ supporting ResNet(79.155%) and Xception(79.945%). Multi-scale & flip test and COCO dataset interface has been finished.

Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet

Amazing Semantic Segmentation on Tensorflow && Keras (include FCN, UNet, SegNet, PSPNet, PAN, RefineNet, DeepLabV3, DeepLabV3+, DenseASPP, BiSegNet)

Reinforcement learning tutorials

Deep Q-learning (DQN) for Multi-agent Reinforcement Learning (RL)

PFENet: Prior Guided Feature Enrichment Network for Few-shot Segmentation (TPAMI).

This repo is the code of paper "DiffusionInst: Diffusion Model for Instance Segmentation" (ICASSP'24).

A basic PyTorch implementation of 'Denoising Diffusion Probabilistic Models'

RGB-Thermal Fusion Network for Semantic Segmentation of Urban Scenes

API to get enormous amount of high resolution satellite images from satellites.pro quickly through multi-threading! create map your own map dataset. Bringing data to Humans.

A collection of pre-trained RL agents using Stable Baselines3

Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)

This repository contains the official code of DS-TransUNet: Dual Swin Transformer U-Net for Medical Image Segmentation

Implementation of various DDPM papers to understand how they work

[WACV 2024] An implementation of MEGANet for polyp segmentation with multi-scale edge-guided attention