Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization.

[ICCV'23] Official PyTorch implementation for paper "Exploring Predicate Visual Context in Detecting Human-Object Interactions"

[CVPR2023] This is an official implementation of paper "DETRs with Hybrid Matching".

Official implementation of BGNN(CVPR 2021)

Official implementation of paper "OED: Towards One-stage End-to-End Dynamic Scene Graph Generation".

An open-source RAG-based tool for chatting with your documents.

🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.

[NeurIPS 2020] Code for the paper "Balanced Meta-Softmax for Long-Tailed Visual Recognition" on LVIS-0.5 dataset

A python wrapper for the Visual Genome API

[CVPR 2022] Official Pytorch code for OW-DETR: Open-world Detection Transformer

[CVPR'2022] SAM-DETR & SAM-DETR++: Official PyTorch Implementation

《动手学大模型Dive into LLMs》系列编程实践教程

[CVPR 2024 Best paper award candidate] EGTR: Extracting Graph from Transformer for Scene Graph Generation

[Unofficial] PyTorch implementation of "Conformer: Convolution-augmented Transformer for Speech Recognition" (INTERSPEECH 2020)

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

[Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications

Causal depthwise conv1d in CUDA, with a PyTorch interface

[IEEE TPAMI-2024] Pair then Relation: Pair-Net for Panoptic Scene Graph Generation

This repository contains code for the paper "Fine-Grained Predicates Learning for Scene Graph Generation (CVPR 2022)".

RelTR: Relation Transformer for Scene Graph Generation: https://arxiv.org/abs/2201.11460v2

Official PyTorch implementation of "Multi-modal Queried Object Detection in the Wild" (accepted by NeurIPS 2023)

A lightweight, scalable, and general framework for visual question answering research

A New Benchmark for Scene Graph Generation, targeting real-world applications

ArXiv Paper Collector for cs.CV

The official Python client for the Huggingface Hub.

An official implementation of the Anchor DETR.

The official code for Devil's on the Edges: Selective Quad Attention for Scene Graph Generation, CVPR2023.