目录
目录README.md

ts-cv

介绍

计算机视觉是人工智能的一个重要分支。本目录下提供了各种经典算法的实现,包括图像分类、目标检测、图像生成、视频分类、关键点检测、图像分割、语义分割、人体姿态估计、3D视觉等各类算法。对于每个模型,我们均提供了模型的定义、训练代码和推理代码,且部分模型提供了train.sh和infer.sh两个脚本,分别对应模型的训练和推理,便于使用者快速上手。

模型分类

图像分类

VGG

AlexNet

ResNet

MobileNetV2

MobileNetV3

ShuffleNetV2

InceptionV3

ResNext50_32x4d

RepVgg

LeNet

GhostNet

PoseNet

SCNet

FaceNet

InsightFace

ReID(BoT)

GAN-MNIST

ViT

DenseNet

MnasNet

SwinTransformer

EfficientnetV2

inceptionv4

xception

RSN

preactresnet

NTS-Net

目标检测

oneflow_yolov3

LDFNet

TCA

oneflow_face

CSRNet

oneflow_detection

Mask_RCNN

图像生成

DCGAN

SRGAN

pix2pix

CycleGAN

pix2pixHD

FastNeuralStyle

TTSR

RFR

视频分类

TSN

AP3D

DLA

FDNet

FMF

BNNVPR

ViewVPR

关键点检测

HRNet

HigherHRNet

RSN

图像分割

LinkNet34

EMANet

HamNet

U-Net

FaceSeg

FODDet

人体姿态估计

FPDNet

KFPN

HPE

SWAHR

HourglassNet

Simple Pose

3D视觉

LoFTR

manhattan_sdf

InvRender

RVL-Dynamic

NeRF

NeuS

目标跟踪

Multi-Object Tracking(Restrack)

    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

©Copyright 2023 CCF 开源发展委员会
Powered by Trustie& IntelliDE 京ICP备13000930号