yankai317

YOLOX

YOLOX is a high-performance anchor-free YOLO. Exceeding yolov3~v5 with ONNX, TensorRT, ncnn, and OpenVINO supported.

cocoapi

COCO API - Dataset @ http://cocodataset.org/

ComfyUI

The most powerful and modular stable diffusion GUI with a graph/nodes interface.

ConvNeXt

Code release for ConvNeXt model

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

fast-reid

SOTA ReID Methods and Toolbox

generative-models

Generative Models by Stability AI

LeViT

LeViT a Vision Transformer in ConvNet's Clothing for Faster Inference

Medical-Image-Processing

MICCAI-LITS2017

liver and liver tumor segmentation using deep learning

MobileViT

Unofficial PyTorch implementation of MobileViT based on paper "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer".

once-for-all

[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment

OOTDiffusion

Official implementation of OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on

PaddleDetection

Object detection and instance segmentation toolkit based on PaddlePaddle.

ProxylessNAS

[ICLR 2019] ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardware.

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

PyTorch-Tutorial

Build your neural network easy and fast

Qbot

[🔥updating ...] 自动量化交易机器人 Qbot is an AI-oriented quantitative investment platform, which aims to realize the potential, empower AI technologies in quantitative investment. https://ufund-me.github.io/Qbot :news: qbot-mini: https://github.com/Charmve/iQuant

qpic

Repo for CVPR2021 paper "QPIC: Query-Based Pairwise Human-Object Interaction Detection with Image-Wide Contextual Information"

rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

sam

SAM: Sharpness-Aware Minimization (PyTorch)

segment

stable-diffusion

A latent text-to-image diffusion model

stable-diffusion-webui

Stable Diffusion web UI

Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Swin-Transformer-Object-Detection

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.

tensorflow-deeplab-v3-plus

DeepLabv3+ built in TensorFlow

tsn-pytorch

Temporal Segment Networks (TSN) in PyTorch

yolov5

YOLOv5 in PyTorch > ONNX > CoreML > TFLite

yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors