Stars
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"
Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.
Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet…
Official codes of DiffBIR: Towards Blind Image Restoration with Generative Diffusion Prior
SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)
Depth Anything V2. A More Capable Foundation Model for Monocular Depth Estimation
[ICLR 2020] Contrastive Representation Distillation (CRD), and benchmark of recent knowledge distillation methods
A simple method to perform semi-supervised learning with limited data.
The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679 and [ICCV2023] DOT: A Distillation-Oriented Trainer https://openaccess.thecvf.com/content…
Fast, differentiable sorting and ranking in PyTorch
整理 pytorch 单机多 GPU 训练方法与原理
RobustBench: a standardized adversarial robustness benchmark [NeurIPS'21 Benchmarks and Datasets Track]
PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection
This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet, vgg, resnet, squeezenet, densenet)
MetaBCI: China’s first open-source platform for non-invasive brain computer interface. The project of MetaBCI is led by Prof. Minpeng Xu from Tianjin University, China.
ICLR21 Tent: Fully Test-Time Adaptation by Entropy Minimization
Focal and Global Knowledge Distillation for Detectors (CVPR 2022)
[CVPR 2022] Official CoTTA Code for our paper Continual Test-Time Domain Adaptation
This repo provides the source code for "Cross-Domain Adaptive Teacher for Object Detection".
A repository and benchmark for online test-time adaptation.