Tao Yuan's starred repositories
emo-visual-data
😜 表情包视觉数据集,使用glm-4v、step-1v的图像解析能力标注。
memory_profiler
Monitor Memory usage of Python code
aesthetic-predictor-v2-5
SigLIP-based Aesthetic Score Predictor
StableCascade
Official Code for Stable Cascade
ComfyUI_Cutoff
cutoff implementation for ComfyUI
CodeFormer
[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer
ImageMagick
🧙♂️ ImageMagick 7
IQA-Dataset
A unified interface for downloading and loading popular Image Quality Assessment (IQA) datasets.
all-of-statistics
Self-study on Larry Wasserman's "All of Statistics"
CompressAI
A PyTorch library and evaluation platform for end-to-end compression research
Real-ESRGAN
Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.
deeplab-pytorch
PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets
pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
mmpretrain
OpenMMLab Pre-training Toolbox and Benchmark
mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
fast_blind_video_consistency
Learning Blind Video Temporal Consistency (ECCV 2018)
focal-frequency-loss
[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis
video-quality
Video quality metrics, reference implementation in python: VIF, SSIM, PSNR, ...
mmdetection
OpenMMLab Detection Toolbox and Benchmark