YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Ultralytics YOLO 🚀

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

State-of-the-art 2D and 3D Face Analysis Project

OCR, layout analysis, reading order, table recognition in 90+ languages

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Face recognition using Tensorflow

Refine high-quality datasets and visual AI models

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)

The project is an official implement of our ECCV2018 paper "Simple Baselines for Human Pose Estimation and Tracking(https://arxiv.org/abs/1804.06208)"

Scaled-YOLOv4: Scaling Cross Stage Partial Network

NVIDIA DeepStream SDK 8.0 / 7.1 / 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training

Brevitas: neural network quantization in PyTorch

Object Detection Metrics. 14 object detection metrics: mean Average Precision (mAP), Average Recall (AR), Spatio-Temporal Tube Average Precision (STT-AP). This project supports different bounding b…

Generate text line images for training deep learning OCR models

This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network.

Python Computer Vision & Video Analytics Framework With Batteries Included

This repository has been deprecated. Suggest to use edgeai-mmdetection in https://github.com/TexasInstruments/edgeai-tensorlab

Python library for analysing faces using PyTorch

yolov7 face detection with landmark

Official implementation of the paper "You Only Need Adversarial Supervision for Semantic Image Synthesis" (ICLR 2021)

Unsupervised Pre-training for Person Re-identification (LUPerson)

Minimal PyTorch implementation of YOLACT.

PyTorch codes for "Real-World Blind Super-Resolution via Feature Matching with Implicit High-Resolution Priors", ACM MM2022 (Oral)

A multi-modal, photo-realistic dataset for online end-to-end scene change detection and more (accepted to IROS2021).

Implementation of the Capsule-Forensics-v2

End-to-end face detection and recognition system using PaddlePaddle.