🌟Change the world, it will become a better place. | 以科研和竞赛为导向的最好的YOLO实践框架!

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Model interpretability and understanding for PyTorch

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

A Large-Scale Dataset for Spinal Vertebrae Segmentation in Computed Tomography

Developing Generalist Foundation Models from a Multimodal Dataset for 3D Computed Tomography

Developing Generalist Foundation Models from a Multimodal Dataset for 3D Computed Tomography

The official repository for "One Model to Rule them All: Towards Universal Segmentation for Medical Images with Text Prompts"

[ECCV 2024] Scene as Gaussians for Vision-Based 3D Semantic Occupancy Prediction

The implementation of "In-Place Scene Labelling and Understanding with Implicit Scene Representation" [ICCV 2021].

MICCAI 2024: Learning 3D Gaussians for Extremely Sparse-View Cone-Beam CT Reconstruction

A Benchmark Dataset: Synthesized DeepLesion for CT Metal Artifact Reduction

unofficial MaskGIT reproduction in PyTorch

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".

A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis

Repository of Vision Transformer with Deformable Attention (CVPR2022) and DAT++: Spatially Dynamic Vision Transformerwith Deformable Attention

Implementation of Deformable Attention in Pytorch from the paper "Vision Transformer with Deformable Attention"

Differentiable fast wavelet transforms in PyTorch with GPU support.

An Open-access Dataset for Liver Lesion Diagnosis on Multi-phase MRI

This code is an official implementation of "DuDoTrans: Dual-Domain Transformer for Sparse-View CT Reconstruction".

Acceptance rates for the major AI conferences

[CVPR 2024] Generalizable Tumor Synthesis - Realistic Synthetic Tumors in Liver, Pancreas, and Kidney

[ICLR 2024 Oral] Supervised Pre-Trained 3D Models for Medical Image Analysis (9,262 CT volumes + 25 annotated classes)

[NeurIPS 2023] AbdomenAtlas 1.0 (5,195 CT volumes + 9 annotated classes)

Helical to fan-beam CT geometry rebinning and differentiable FBP of DICOM-CT-PD projections

A large-scale dataset of both raw MRI measurements and clinical MRI images.

FcaNet: Frequency Channel Attention Networks

Sharpness-aware Low Dose CT Denoising Using Conditional Generative Adversarial Network