Stars
Pytorch Pedestrian Attribute Recognition: A strong PyTorch baseline for pedestrian attribute recognition and multi-label classification.
A Semantic Controllable Self-Supervised Learning Framework to learn general human representations from massive unlabeled human images, which can benefit downstream human-centric tasks to the maximu…
Learn to build a Fashion Classifier using Convolutional Neural Network.
Uncertainty Quantification via Spatial-Temporal Tweedie Model for Zero-inflated and Long-tail Travel Demand Prediction
Traffic accident prediction using graph neural networks
My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entr…
Code release for ActionFormer (ECCV 2022)
A dataset for traffic accident analysis in the US
Concept Modeling: Topic Modeling on Images and Text
Website for storing flight information, rendering paths on a zoomable world map and calculating statistics, with plenty of free airline, airport and route data.
[Arxiv2020] The code for our paper 《Self-Supervised Temporal-Discriminative Representation Learning for Video Action Recognition》 https://arxiv.org/abs/2008.02129
OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
[CVPR 2023] Official Implementation of X-Decoder for generalized decoding for pixel, image and language
Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]
[ICCV 2023] Official implementation of the paper "A Simple Framework for Open-Vocabulary Segmentation and Detection"
BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
We write your reusable computer vision tools. 💜
Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
[NeurIPS 2023] Fine-Grained Cross-View Geo-Localization Using a Correlation-Aware Homography Estimator
Official repository for TransGeo: Transformer Is All You Need for Cross-view Image Geo-localization
PyTorch implementations of Generative Adversarial Networks.
Image-to-image translation with conditional adversarial nets
Models and examples built with TensorFlow