# ts-cv **Repository Path**: corner2007/ts-cv ## Basic Information - **Project Name**: ts-cv - **Description**: 计算机视觉是人工智能的一个重要分支。本仓库中包含了各种图像相关的算法，均基于oneflow框架实现。 - **Primary Language**: Unknown - **License**: Not specified - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 7 - **Created**: 2025-07-18 - **Last Updated**: 2025-07-18 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # ts-cv ## 介绍 计算机视觉是人工智能的一个重要分支。本目录下提供了各种经典算法的实现，包括图像分类、目标检测、图像生成、视频分类、关键点检测、图像分割、语义分割、人体姿态估计、3D视觉等各类算法。对于每个模型，我们均提供了模型的定义、训练代码和推理代码，且部分模型提供了train.sh和infer.sh两个脚本，分别对应模型的训练和推理，便于使用者快速上手。 ## 模型分类 ### **图像分类** [VGG ](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/vgg) [AlexNet ](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/alexnet) [ResNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/resnet50) [MobileNetV2](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/mobilenetv2) [MobileNetV3](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/mobilenetv3) [ShuffleNetV2](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/shufflenetv2) [InceptionV3](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/inception_v3) [ResNext50_32x4d](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/resnext50_32x4d) [RepVgg](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/repvgg) [LeNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/LeNet) [GhostNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/ghostnet) [PoseNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/poseNet) [SCNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/scnet) [FaceNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/facenet) [InsightFace ](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/InsightFace) [ReID(BoT)]( https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/reid/BoT) [GAN-MNIST ](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/GAN-MNIST) [ViT](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/ViT) [DenseNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/densenet) [MnasNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/mnasnet) [SwinTransformer ](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/swin_transformer) [EfficientnetV2](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/EfficientnetV2) [inceptionv4](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/inceptionv4) [xception](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/xception) [RSN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/RSN) [preactresnet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/preactresnet) [NTS-Net](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/NTS-Net-Oneflow) ### **目标检测** [oneflow_yolov3](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/oneflow_yolov3) [LDFNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/LDF) [TCA](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/TCA) [oneflow_face](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/oneflow_face) [CSRNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/detection/CSRNet) [oneflow_detection](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/oneflow_detection) [Mask_RCNN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Mask_RCNN) ### **图像生成** [DCGAN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/gan/DCGAN) [SRGAN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/gan/SRGAN) [pix2pix](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/gan/pix2pix) [CycleGAN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/gan/cycleGAN) [pix2pixHD](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/pix2pix-hd) [FastNeuralStyle](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/style_transform/fast_neural_style) [TTSR](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/TTSR) [RFR](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/RFR-Inpainting) ### **视频分类** [TSN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/video/TSN) [AP3D]( https://gitee.com/zhijiangtianshu/ts-cv/tree/master/AP3D) [DLA](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/classification/image/DLA) [FDNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/FDNet_oneflow/FDNet_oneflow) [FMF](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/FMF_oneflow/FMF_oneflow) [BNNVPR]( https://gitee.com/zhijiangtianshu/ts-cv/tree/master/BNNVPR_oneflow/BNNVPR_oneflow) [ViewVPR](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/ViewVPR_oneflow/ViewVPR_oneflow) ### **关键点检测** [HRNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Hrnet) [HigherHRNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/HigherHRNet) [RSN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/RSN) ### **图像分割** [LinkNet34](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/LinkNet34) [EMANet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/EMANet) [HamNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/HamNet) [U-Net](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/segmentation/U-Net) [FaceSeg](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/segmentation/FaceSeg) [FODDet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Visions/segmentation/FODDet) ### **人体姿态估计** [FPDNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/fast-human-pose-estimation(FPDNet)) [KFPN](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/KFPN) [HPE](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/HPE) [SWAHR](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/SWAHR) [HourglassNet](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/stacked_hourglass) [Simple Pose](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/posenet-rc) ### **3D视觉** [LoFTR](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/LoFTR) [manhattan_sdf](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/manhattan_sdf) [InvRender](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/InvRender) [RVL-Dynamic](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/RVL-Dynamic) [NeRF](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/NeRF) [NeuS](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/NeuS) ### 目标跟踪 [Multi-Object Tracking（Restrack）](https://gitee.com/zhijiangtianshu/ts-cv/tree/master/Multi-Object%20Tracking(Restrack))