C. L. Wang's starred repositories
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
deep-learning-for-image-processing
deep learning for image processing including classification and object-detection etc.
dalle-mini
DALL·E Mini - Generate images from a text prompt
Swin-Transformer
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
pytorch-grad-cam
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
pytorch-summary
Model summary in PyTorch similar to `model.summary()` in Keras
mmclassification
OpenMMLab Image Classification Toolbox and Benchmark
pytorch_image_classification
PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet
transfiner
Mask Transfiner for High-Quality Instance Segmentation, CVPR 2022
DCNv2_latest
DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
pycytominer
Python package for processing image-based profiling data
spatial-VAE
Source code for "Explicitly disentangling image content from translation and rotation with spatial-VAE" - NeurIPS 2019
rxrx1-utils
Starter code for the CellSignal NeurIPS 2019 competition.
TrainingDynamics
Compute training dynamics, plot data cartography, analysing data quality...