Yidan-Zhang's repositories
RePaint-Inpainting-using-Denoising-Diffusion-Probabilistic-Models
Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022
awesome-semantic-segmentation-pytorch
Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
BGNet
Boundary-Guided Camouflaged Object Detection
CS-Notes
:books: 技术面试必备基础知识、Leetcode、计算机操作系统、计算机网络、系统设计
CVPR-2022-papers
CVPR 2022 papers with code (论文及代码)
CvPytorch
CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.
Deep-Reinforcement-Learning-Algorithms-with-PyTorch
PyTorch implementations of deep reinforcement learning algorithms and environments
Deep-reinforcement-learning-with-pytorch
PyTorch implementation of DQN, AC, ACER, A2C, A3C, PG, DDPG, TRPO, PPO, SAC, TD3 and ....
Graphormer
Graphormer is a deep learning package that allows researchers and developers to train custom models for molecule modeling tasks. It aims to accelerate the research and application in AI for molecule science, such as material design, drug discovery, etc.
iaanet-Interior-Attention-Aware-Network-for-Infrared-Small-Target-Detection
(TGRS 2022) Paper: Interior Attention-Aware Network for Infrared Small Target Detection
ICNet-pytorch
ICNet implemented by pytorch, for real-time semantic segmentation on high-resolution images, mIOU=71.0 on cityscapes, single inference time is 19ms, FPS is 52.6.
MIRST_dataset
A large-scale multi-scene dataset constructed for single-frame infrared small target detection.
mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
PaddleSeg
Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, etc.
PyQt5-YOLOv5
YOLOv5检测界面-PyQt5实现
pytorch-grad-cam
Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM
PyTorch-Networks
Pytorch implementation of cnn network
Reinforcement-Implementation
Implementation of benchmark RL algorithms
SETR
[CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers
SNL_ICCV2021
Official implementation of the paper ``Unifying Nonlocal Blocks for Neural Networks'' (ICCV'21)
SPFNet
SPFNet
spinningup
An educational resource to help anyone learn deep reinforcement learning.
UCTransNet
Implementation of our AAAI'22 work: "UCTransNet: Rethinking the Skip Connections in U-Net from a Channel-wise Perspective with Transformer".
UTNet
Official implementation of UTNet: A Hybrid Transformer Architecture for Medical Image Segmentation
YOLOv5-Multibackbone-Compression
YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming) and Quantization (MQBench) Compression Tool Box.