zeyuli1990

MAgent

A Platform for Many-Agent Reinforcement Learning

MARL-Papers

Paper list of multi-agent reinforcement learning (MARL)

HARL

Official implementation of HARL algorithms based on PyTorch.

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Diffusion-Models-Papers-Survey-Taxonomy

Diffusion model papers, survey, and taxonomy

fft-conv-pytorch

Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.

PyTorch_YOLO-Family

Paper-Picture-Writing-Code

MLNLP: Paper Picture Writing Code

MTBook

《机器翻译：基础与模型》肖桐 朱靖波 著 - Machine Translation: Foundations and Models

lemniscate.pytorch

Unsupervised Feature Learning via Non-parametric Instance Discrimination

google-research

Google Research

EasyCV

An all-in-one toolkit for computer vision

YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

BalancedMetaSoftmax

[NeurIPS 2020] Code for the paper "Balanced Meta-Softmax for Long-Tailed Visual Recognition"

HRSID

HRSID: high resolution sar images dataset for ship detection, semantic segmentation, and instance segmentation tasks.

Pretrained-IPT

Machine-Learning-Notes

白板推导系列课程笔记 初版

PyTorch-Pretrained-ViT

Vision Transformer (ViT) in PyTorch

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

detr

End-to-End Object Detection with Transformers

MixNMatch

Pytorch implementation of MixNMatch

TransGAN

[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang

iSketchNFill

Software that can autocomplete sketches as the user starts drawing.

EdgeGAN

Code for paper "SketchyCOCO: Image Generation from Freehand Scene Sketches" (CVPR 2020)

SketchyCOCO

SketchyCOCO dataset (CVPR 2020)

AttenNorm

Attentive Normalization for Conditional Image Generation

stylegan2

StyleGAN2 - Official TensorFlow Implementation

NN-SVG

Publication-ready NN-architecture schematics.

SciencePlots

Matplotlib styles for scientific plotting