Stars
Deformable ConvNets V2 (DCNv2) in PyTorch
Official implementation of the CVPR 2024 paper ViT-CoMer: Vision Transformer with Convolutional Multi-scale Feature Interaction for Dense Predictions.
Implementation of ACER and A3C in PyTorch.
Paper list and datasets for industrial image anomaly/defect detection (updating). 工业异常/瑕疵检测论文及数据集检索库(持续更新)。
Official implementation of CVPR'24 paper 'Toward Generalist Anomaly Detection via In-context Residual Learning with Few-shot Sample Prompts'.
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
[CVPR 2023] Official implementation of the paper "Lite DETR : An Interleaved Multi-Scale Encoder for Efficient DETR"
Official implementation of "Segment Any Anomaly without Training via Hybrid Prompt Regularization (SAA+)".
[ICRA'23] BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird's-Eye View Representation
[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Official code for Conformer: Local Features Coupling Global Representations for Visual Recognition
BigDetection: A Large-scale Benchmark for Improved Object Detector Pre-training
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
Training a number of models to classify Fighting vs Non Fighting Images.
pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...
This is a simple demo to demonstrate how to perfrom GradCAM on the detectron2 models and it's API
Coldestadam / DynamicHead
Forked from microsoft/DynamicHeadPyTorch ,ONNX and TensorRT implementation of YOLOv4
Official Pytorch implementation of CutMix regularizer
An implementation of feature visualizer for pytorch tensor, based on Matplotlib.
Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object Detection, NeurIPS2020
Multiple Object Tracker, Based on Hungarian algorithm + Kalman filter.
A Tensorflow implementation of Spatial Transformer Networks.
Partial Order Pruning: for Best Speed/Accuracy Trade-off in Neural Architecture Search
AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.