# DeepSparkHub **Repository Path**: LYCSMS_ms_admin/deepsparkhub ## Basic Information - **Project Name**: DeepSparkHub - **Description**: DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领域，支持主流市场智能计算落地应用场景。 - **Primary Language**: Python - **License**: Apache-2.0 - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 59 - **Created**: 2024-04-30 - **Last Updated**: 2024-10-11 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # DeepSparkHub DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领域，支持主流市场智能计算场景，包括智慧城市、数字个人、医疗、教育、通信、能源等多个领域。 ## 模型列表 - Computer Vision - [Classification](#classification) - [Face Detection](#face-detection) - [Face Recognition](#face-recognition) - [Instance Segmentation](#instance-segmentation) - [Knowledge Distillation](#knowledge-distillation) - [Network Pruning](#network-pruning) - [Object Detection](#object-detection) - [3D Object Detection](#3d-object-detection) - [OCR](#ocr) - [Point Cloud](#point-cloud) - [Pose Estimation](#pose-estimation) - [Self-Supervised Learning](#self-supervised-learning) - [Semantic Segmentation](#semantic-segmentation) - [Super Resolution](#super-resolution) - [Tracking](#tracking) - [Traffic Forecast](#traffic-forecast) - Graph Neural Network (GNN) - [Graph Attention](#graph-attention) - [Node Classification](#node-classification) - [Text Classification](#text-classification) - High Performance Computing (HPC) - [Molecular Dynamics](#molecular-dynamics) - [Multimodal](#multimodal) - Natural Language Processing (NLP) - [Cloze Test](#cloze-test) - [Dialogue Generation](#dialogue-generation) - [Language Modeling](#language-modeling) - [Large Language Model (LLM)](#large-language-model-llm) - [Text Correction](#text-correction) - [Translation](#translation) - Recommendation - [Collaborative Filtering](#collaborative-filtering) - [Click Through Rate](#click-through-rate) - [Reinforcement Learning](#reinforcement-learning) - Speech - [Speech Recognition](#speech-recognition) - [Speech Synthesis](#speech-synthesis) - [3D Reconstruction](#3d-reconstruction) -------- ### Computer Vision #### Classification 模型名称 | 框架 | 数据集 -------- | ------ | ---- [ACmix](cv/classification/acmix/pytorch/README.md) | PyTorch | ImageNet [ACNet](cv/classification/acnet/pytorch/README.md) | PyTorch | ImageNet [AlexNet](cv/classification/alexnet/pytorch/README.md) | PyTorch | ImageNet [AlexNet](cv/classification/alexnet/tensorflow/README.md) | TensorFlow | ImageNet [BYOL](cv/classification/byol/pytorch/README.md) | PyTorch | ImageNet [CBAM](cv/classification/cbam/pytorch/README.md)  | PyTorch | ImageNet [ConvNext](cv/classification/convnext/pytorch/README.md) | PyTorch | ImageNet [CspDarknet53](cv/classification/cspdarknet53/pytorch/README.md)  | PyTorch | ImageNet [DenseNet](cv/classification/densenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet [DenseNet](cv/classification/densenet/pytorch/README.md) | PyTorch | ImageNet [DPN92](cv/classification/dpn92/pytorch/README.md) | PyTorch | ImageNet [DPN107](cv/classification/dpn107/pytorch/README.md) | PyTorch | ImageNet [ECA-MobileNetV2](cv/classification/eca_mobilenet_v2/pytorch/README.md) | PyTorch | ImageNet [ECA-ResNet152](cv/classification/eca_resnet152/pytorch/README.md) | PyTorch | ImageNet [EfficientNetB0](cv/classification/efficientnet_b0/paddlepaddle/README.md) | PaddlePaddle | ImageNet [EfficientNetB4](cv/classification/efficientnet_b4/pytorch/README.md) | PyTorch | ImageNet [FasterNet](cv/classification/fasternet/pytorch/README.md)  | PyTorch | ImageNet [GoogLeNet](cv/classification/googlenet/pytorch/README.md) | PyTorch | ImageNet [GoogLeNet](cv/classification/googlenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet [InceptionV3](cv/classification/inceptionv3/mindspore/README.md) | MindSpore | ImageNet [InceptionV3](cv/classification/inceptionv3/pytorch/README.md) | PyTorch | ImageNet [InceptionV3](cv/classification/inceptionv3/tensorflow/README.md) | TensorFlow | ImageNet [InceptionV4](cv/classification/inceptionv4/pytorch/README.md) | PyTorch | ImageNet [InternImage](cv/classification/internimage/pytorch/README.md) | PyTorch | ImageNet [LeNet](cv/classification/lenet/pytorch/README.md) | PyTorch | ImageNet [MobileNetV2](cv/classification/mobilenetv2/pytorch/README.md) | PyTorch | ImageNet [MobileNetV3](cv/classification/mobilenetv3/mindspore/README.md) | MindSpore | ImageNet [MobileNetV3](cv/classification/mobilenetv3/pytorch/README.md) | PyTorch | ImageNet [MobileNetV3](cv/classification/mobilenetv3/paddlepaddle/README.md) | PaddlePaddle | ImageNet [MobileNetV3_Large1.0](cv/classification/mobilenetv3_large_x1_0/paddlepaddle/README.md) | PaddlePaddle | ImageNet [MobileOne](cv/classification/mobileone/pytorch/README.md) | PyTorch | ImageNet [MoCoV2](cv/classification/mocov2/pytorch/README.md) | PyTorch | ImageNet [PP-LCNet](cv/classification/pp-lcnet/paddlepaddle/README.md) | PaddlePaddle | ImageNet [RepMLP](cv/classification/repmlp/pytorch/README.md) | PyTorch | ImageNet [RepVGG](cv/classification/repvgg/pytorch/README.md) | PyTorch | ImageNet [RepVGG](cv/classification/repvgg/paddlepaddle/README.md) | PaddlePaddle | ImageNet [RepViT](cv/classification/repvit/pytorch/README.md) | PyTorch | ImageNet [Res2Net50_14w_8s](cv/classification/Res2Net50_14w_8s/paddlepaddle/README.md) | PaddlePaddle | ImageNet [ResNeSt14](cv/classification/resnest14/pytorch/README.md) | PyTorch | ImageNet [ResNeSt50](cv/classification/resnest50/pytorch/README.md) | PyTorch | ImageNet [ResNeSt50](cv/classification/resnest50/paddlepaddle/README.md) | PaddlePaddle | ImageNet [ResNeSt101](cv/classification/resnest101/pytorch/README.md) | PyTorch | ImageNet [ResNeSt269](cv/classification/resnest269/pytorch/README.md) | PyTorch | ImageNet [ResNet18](cv/classification/resnet18/pytorch/README.md) | PyTorch | ImageNet [ResNet50](cv/classification/resnet50/pytorch/README.md) | PyTorch | ImageNet [ResNet50](cv/classification/resnet50/paddlepaddle/README.md) | PaddlePaddle | ImageNet [ResNet50](cv/classification/resnet50/tensorflow/README.md) | TensorFlow | ImageNet [ResNet101](cv/classification/resnet101/pytorch/README.md) | PyTorch | ImageNet [ResNet152](cv/classification/resnet152/pytorch/README.md) | PyTorch | ImageNet [ResNeXt50_32x4d](cv/classification/resnext50_32x4d/mindspore/README.md) | MindSpore | ImageNet [ResNeXt50_32x4d](cv/classification/resnext50_32x4d/pytorch/README.md) | PyTorch | ImageNet [ResNeXt101_32x8d](cv/classification/resnext101_32x8d/pytorch/README.md) | PyTorch | ImageNet [SE_ResNet50_vd](cv/classification/se_resnet50_vd/paddlepaddle/README.md) | PaddlePaddle | ImageNet [SEResNeXt](cv/classification/seresnext/pytorch/README.md) | PyTorch | ImageNet [ShuffleNetV2](cv/classification/shufflenetv2/paddlepaddle/README.md) | PaddlePaddle | ImageNet [ShuffleNetV2](cv/classification/shufflenetv2/pytorch/README.md) | PyTorch | ImageNet [SqueezeNet](cv/classification/squeezenet/pytorch/README.md) | PyTorch | ImageNet [Swin Transformer](cv/classification/swin_transformer/paddlepaddle/README.md) | PaddlePaddle | ImageNet [Swin Transformer](cv/classification/swin_transformer/pytorch/README.md) | PyTorch | ImageNet [VGG16](cv/classification/vgg/paddlepaddle/README.md) | PaddlePaddle | ImageNet [VGG16](cv/classification/vgg/pytorch/README.md) | PyTorch | ImageNet [VGG16](cv/classification/vgg/tensorflow/README.md) | TensorFlow | ImageNet [Wave-MLP](cv/classification/wavemlp/pytorch/README.md) | PyTorch | ImageNet [Wide_ResNet101_2](cv/classification/wide_resnet101_2/pytorch/README.md) | PyTorch | ImageNet [Xception](cv/classification/xception/paddlepaddle/README.md) | PaddlePaddle | ImageNet [Xception](cv/classification/xception/pytorch/README.md) | PyTorch | ImageNet #### Face Detection 模型名称 | 框架 | 数据集 -------- | ------ | ---- [RetinaFace](cv/face/retinaface/pytorch/README.md) | PyTorch | WiderFace #### Face Recognition 模型名称 | 框架 | 数据集 -------- | ------ | ---- [ArcFace](cv/face/arcface/pytorch/README.md)  | PyTorch | CASIA-WebFaces&LFW [BlazeFace](cv/face/blazeface/paddlepaddle/README.md)  | PaddlePaddle | WIDER-FACE [CosFace](cv/face/cosface/pytorch/README.md)  | PyTorch | CASIA-WebFaces&LFW [FaceNet](cv/face/facenet/pytorch/README.md)  | PyTorch | CASIA-WebFaces&LFW [FaceNet](cv/face/facenet/tensorflow/README.md) | TensorFlow | CASIA-WebFaces&LFW #### Instance Segmentation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [SOLO](cv/instance_segmentation/SOLO/pytorch/README.md) | PyTorch | COCO [SOLOv2](cv/detection/solov2/paddlepaddle/README.md) | PaddlePaddle | COCO [SOLOv2](cv/instance_segmentation/solov2/pytorch/README.md) | PyTorch | COCO [YOLACT++](cv/instance_segmentation/yolact/pytorch/README.md) | PyTorch | COCO #### Image Generation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [DCGAN](cv/image_generation/dcgan/mindspore/README.md) | MindSpore | ImageNet [Pix2Pix](cv/image_generation/Pix2pix/paddlepaddle/README.md) | PaddlePaddle | facades #### Knowledge Distillation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [CWD](cv/distiller/CWD/pytorch/README.md)  | PyTorch | Cityscapes [RKD](cv/distiller/RKD/pytorch/README.md)  | PyTorch | CUB-200-2011 [WSLD](cv/distiller/WSLD/pytorch/README.md) | PyTorch | ImageNet #### Network Pruning 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Network Slimming](cv/Pruning/Network-Slimming/pytorch/README.md)  | PyTorch | CIFAR-10/100 #### Object Detection 模型名称 | 框架 | 数据集 -------- | ------ | ---- [ATSS](cv/detection/atss_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO [AutoAssign](cv/detection/autoassign/pytorch/README.md) | PyTorch | COCO [Cascade R-CNN](cv/detection/cascade_rcnn_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO [CenterMask2](cv/detection/centermask2/pytorch/README.md) | PyTorch | COCO [CenterNet](cv/detection/centernet/pytorch/README.md) | PyTorch | COCO [CenterNet](cv/detection/centernet/paddlepaddle/README.md) | PaddlePaddle | COCO [Co-DETR](cv/detection/co-detr/pytorch/README.md) | PyTorch | COCO [CornerNet](cv/detection/cornernet_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO [DCNV2](cv/detection/dcnv2_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO [DeepSORT](cv/tracking/deep_sort/pytorch/README.md) | PyTorch | Market-1501 [DETR](cv/detection/detr/paddlepaddle/README.md) | PaddlePaddle | COCO [Faster R-CNN](cv/detection/fasterrcnn/pytorch/README.md) | PyTorch | COCO [FCOS](cv/detection/fcos/paddlepaddle/README.md) | PaddlePaddle | COCO [FCOS](cv/detection/fcos/pytorch/README.md) | PyTorch | COCO [Mask R-CNN](cv/detection/maskrcnn/pytorch/README.md) | PyTorch | COCO [Mask R-CNN](cv/detection/maskrcnn/paddlepaddle/README.md) | PaddlePaddle | COCO [OC_SORT](cv/detection/oc_sort/paddlepaddle/README.md) | PaddlePaddle | MOT17 [Oriented RepPoints](cv/detection/oriented_reppoints/pytorch/README.md) | PyTorch | DOTA [PP-PicoDet](cv/detection/picodet/paddlepaddle/README.md) | PaddlePaddle | COCO [PP-YOLOE](cv/detection/pp-yoloe/paddlepaddle/README.md) | PaddlePaddle | COCO [PP-YOLOE+](cv/detection/pp_yoloe+/paddlepaddle/README.md) | PaddlePaddle | COCO [PVANet](cv/detection/pvanet/pytorch/README.md) | PyTorch | COCO [RepPoints](cv/detection/reppoints_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO [RetinaNet](cv/detection/retinanet/pytorch/README.md) | PyTorch | COCO [RetinaNet](cv/detection/retinanet/paddlepaddle/README.md) | PaddlePaddle | COCO [RT-DETR](cv/detection/rt-detr/pytorch/README.md) | PyTorch | COCO [RTMDet](cv/detection/rtmdet/pytorch/README.md) | PyTorch | COCO [SSD](cv/detection/ssd/pytorch/README.md) | PyTorch | COCO [SSD](cv/detection/ssd/paddlepaddle/README.md) | PaddlePaddle | COCO [SSD](cv/detection/ssd/tensorflow/README.md) | TensorFlow | VOC [SSD](cv/detection/ssd/mindspore/README.md) | MindSpore | COCO [YOLOF](cv/detection/yolof/pytorch/README.md) | PyTorch | COCO [YOLOv3](cv/detection/yolov3/pytorch/README.md) | PyTorch | COCO [YOLOv3](cv/detection/yolov3/paddlepaddle/README.md) | PaddlePaddle | COCO [YOLOv3](cv/detection/yolov3/tensorflow/README.md) | TensorFlow | VOC [YOLOv5](cv/detection/yolov5/paddlepaddle/README.md) | PaddlePaddle | COCO [YOLOv5](cv/detection/yolov5/pytorch/README.md) | PyTorch | COCO [YOLOv6](cv/detection/yolov6/pytorch/README.md)  | PyTorch | COCO [YOLOv7](cv/detection/yolov7/pytorch/README.md) | PyTorch | COCO [YOLOv8](cv/detection/yolov8/pytorch/README.md)  | PyTorch | COCO [YOLOv9](cv/detection/yolov9/pytorch/README.md)  | PyTorch | COCO [YOLOv10](cv/detection/yolov10/pytorch/README.md)  | PyTorch | COCO #### 3D Object Detection 模型名称 | 框架 | 数据集 -------- | ------ | ---- [BEVFormer](cv/3d_detection/BEVFormer/pytorch/README.md) | PyTorch | nuScenes&CAN bus [CenterPoint](cv/3d_detection/centerpoint/pytorch/README.md) | PyTorch | nuScenes [PAConv](cv/3d_detection/PAConv/pytorch/README.md) | PyTorch | S3DIS [Part-A2-Anchor](cv/3d_detection/part_a2_anchor/pytorch/README.md) | PyTorch | KITTI [Part-A2-Free](cv/3d_detection/part_a2_free/pytorch/README.md) | PyTorch | KITTI [PointNet++](cv/3d_detection/pointnet2/pytorch/mmdetection3d/README.md)  | PyTorch | S3DIS [PointPillars](cv/3d_detection/pointpillars/pytorch/README.md) | PyTorch | KITTI [PointRCNN](cv/3d_detection/pointrcnn/pytorch/README.md) | PyTorch | KITTI [PointRCNN-IoU](cv/3d_detection/pointrcnn_iou/pytorch/README.md) | PyTorch | KITTI [SECOND](cv/3d_detection/second/pytorch/README.md) | PyTorch | KITTI [SECOND-IoU](cv/3d_detection/second_iou/pytorch/README.md) | PyTorch | KITTI #### OCR 模型名称 | 框架 | 数据集 -------- | ------ | ---- [CRNN](cv/ocr/crnn/mindspore/README.md) | MindSpore | OCR_Recog [CRNN](cv/ocr/crnn/paddlepaddle/README.md) | PaddlePaddle | LMDB [DBNet](cv/ocr/dbnet/pytorch/README.md) | PyTorch | ICDAR2015 [DBNet++](cv/ocr/dbnetpp/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 [DBNet++](cv/ocr/dbnetpp/pytorch/README.md) | PyTorch | ICDAR2015 [PP-OCR-DB](cv/ocr/pp-ocr-db/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 [PP-OCR-EAST](cv/ocr/pp-ocr-east/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 [PSE](cv/ocr/pse/paddlepaddle/README.md) | PaddlePaddle | OCR_Recog [SAR](cv/ocr/sar/pytorch/README.md) | PyTorch | OCR_Recog [SAST](cv/ocr/sast/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 [SATRN](cv/ocr/satrn/pytorch/base/README.md) | PyTorch | OCR_Recog #### Point Cloud 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Point-BERT](cv/point_cloud/Point-BERT/pytorch/README.md) | PyTorch | ShapeNet55 & processed ModelNet #### Pose Estimation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [AlphaPose](cv/pose/alphapose/pytorch/README.md)  | PyTorch | COCO [HRNet](cv/pose/hrnet/pytorch/README.md) | PyTorch | COCO [HRNet-W32](cv/pose/hrnet/paddlepaddle/README.md) | PaddlePaddle | COCO [OpenPose](cv/pose/openpose/mindspore/README.md) | MindSpore | COCO #### Self-Supervised Learning 模型名称 | 框架 | 数据集 -------- | ------ | ---- [MAE](cv/self_supervised_learning/MAE/pytorch/README.md)  | PyTorch | ImageNet #### Semantic Segmentation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [3D-UNet](cv/semantic_segmentation/unet3d/pytorch/README.md) | PyTorch | kits19 [APCNet](cv/semantic_segmentation/apcnet/pytorch/README.md) | PyTorch | Cityscapes [Attention U-net](cv/semantic_segmentation/att_unet/pytorch/README.md)  | PyTorch | Cityscapes [BiSeNet](cv/semantic_segmentation/bisenet/pytorch/README.md) | PyTorch | COCO [BiSeNetV2](cv/semantic_segmentation/bisenetv2/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [BiSeNetV2](cv/semantic_segmentation/bisenetv2/pytorch/README.md) | PyTorch | Cityscapes [CGNet](cv/semantic_segmentation/cgnet/pytorch/README.md) | PyTorch | COCO [ContextNet](cv/semantic_segmentation/contextnet/pytorch/README.md) | PyTorch | COCO [DabNet](cv/semantic_segmentation/dabnet/pytorch/README.md) | PyTorch | COCO [DANet](cv/semantic_segmentation/danet/pytorch/README.md) | PyTorch | COCO [DDRnet](cv/semantic_segmentation/ddrnet/pytorch/README.md)  | PyTorch | Cityscapes [DeepLabV3](cv/semantic_segmentation/deeplabv3/pytorch/README.md) | PyTorch | COCO [DeepLabV3](cv/semantic_segmentation/deeplabv3/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [DeepLabV3](cv/semantic_segmentation/deeplabv3/mindspore/README.md) | MindSpore | VOC [DeepLabV3+](cv/semantic_segmentation/deeplabv3plus/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [DeepLabV3+](cv/semantic_segmentation/deeplabv3plus/tensorflow/README.md) | TensorFlow | Cityscapes [DenseASPP](cv/semantic_segmentation/denseaspp/pytorch/README.md) | PyTorch | COCO [DFANet](cv/semantic_segmentation/dfanet/pytorch/README.md) | PyTorch | COCO [DNLNet](cv/semantic_segmentation/dnlnet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [DUNet](cv/semantic_segmentation/dunet/pytorch/README.md) | PyTorch | COCO [EncNet](cv/semantic_segmentation/encnet/pytorch/README.md) | PyTorch | COCO [ENet](cv/semantic_segmentation/enet/pytorch/README.md) | PyTorch | COCO [ERFNet](cv/semantic_segmentation/erfnet/pytorch/README.md) | PyTorch | COCO [ESPNet](cv/semantic_segmentation/espnet/pytorch/README.md) | PyTorch | COCO [FastFCN](cv/semantic_segmentation/fastfcn/paddlepaddle/README.md) | PyTorch | ADE20K [FastSCNN](cv/semantic_segmentation/fastscnn/pytorch/README.md) | PyTorch | COCO [FCN](cv/semantic_segmentation/fcn/pytorch/README.md) | PyTorch | COCO [FPENet](cv/semantic_segmentation/fpenet/pytorch/README.md) | PyTorch | COCO [GCNet](cv/semantic_segmentation/gcnet/pytorch/README.md) | PyTorch | Cityscapes [HardNet](cv/semantic_segmentation/hardnet/pytorch/README.md) | PyTorch | COCO [ICNet](cv/semantic_segmentation/icnet/pytorch/README.md) | PyTorch | COCO [LedNet](cv/semantic_segmentation/lednet/pytorch/README.md) | PyTorch | COCO [LinkNet](cv/semantic_segmentation/linknet/pytorch/README.md) | PyTorch | COCO [Mask2Former](cv/semantic_segmentation/Mask2Former/pytorch/README.md) | PyTorch | Cityscapes [MobileSeg](cv/semantic_segmentation/mobileseg/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [OCNet](cv/semantic_segmentation/ocnet/pytorch/README.md) | PyTorch | COCO [OCRNet](cv/semantic_segmentation/ocrnet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [OCRNet](cv/semantic_segmentation/ocrnet/pytorch/README.md) | PyTorch | Cityscapes [PP-HumanSegV1](cv/semantic_segmentation/pp_humansegv1/paddlepaddle/README.md) | PaddlePaddle | PP-HumanSeg14K [PP-HumanSegV2](cv/semantic_segmentation/pp_humansegv2/paddlepaddle/README.md) | PaddlePaddle | PP-HumanSeg14K [PP-LiteSeg](cv/semantic_segmentation/pp_liteseg/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [PSANet](cv/semantic_segmentation/psanet/pytorch/README.md) | PyTorch | COCO [RefineNet](cv/semantic_segmentation/refinenet/pytorch/README.md) | PyTorch | COCO [SegNet](cv/semantic_segmentation/segnet/pytorch/README.md) | PyTorch | COCO [STDC](cv/semantic_segmentation/stdc/paddlepaddle/README.md)  | PaddlePaddle | Cityscapes [STDC](cv/semantic_segmentation/stdc/pytorch/README.md)  | PyTorch | Cityscapes [UNet](cv/semantic_segmentation/unet/pytorch/README.md) | PyTorch | COCO [UNet](cv/semantic_segmentation/unet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [UNet++](cv/semantic_segmentation/unet++/pytorch/README.md)  | PyTorch | DRIVE [VNet](cv/semantic_segmentation/vnet/tensorflow/README.md) | TensorFlow | Hippocampus #### Super Resolution 模型名称 | 框架 | 数据集 -------- | ------ | ---- [basicVSR++](cv/super_resolution/basicVSR++/pytorch/README.md) | PyTorch | REDS [basicVSR](cv/super_resolution/basicVSR/pytorch/README.md) | PyTorch | REDS [ESRGAN](cv/super_resolution/esrgan/pytorch/README.md) | PyTorch | DIV2K [LIIF](cv/super_resolution/liif/pytorch/README.md) | PyTorch | DIV2K [RealBasicVSR](cv/super_resolution/real_basicVSR/pytorch/README.md) | PyTorch | REDS [TTSR](cv/super_resolution/ttsr/pytorch/README.md) | PyTorch | CUFED [TTVSR](cv/super_resolution/ttvsr/pytorch/README.md) | PyTorch | REDS #### Tracking 模型名称 | 框架 | 数据集 -------- | ------ | ---- [ByteTrack](cv/tracking/bytetrack/paddlepaddle/README.md) | PaddlePaddle | MOT17 [FairMOT](cv/tracking/fairmot/pytorch/README.md) | PyTorch | MOT17 #### Traffic Forecast 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Graph WaveNet](cv/traffic_forecast/graph_wavenet/pytorch/README.md) | PyTorch | METR-LA & PEMS-BAY ### GNN #### Graph Attention 模型名称 | 框架 | 数据集 -------- | ------ | ---- [GAT](gnn/graph_attention/gat/paddlepaddle/README.md) | PaddlePaddle | CORA #### Node Classification 模型名称 | 框架 | 数据集 -------- | ------ | ---- [GraphSAGE](gnn/node_classification/graphsage/paddlepaddle/README.md) | PaddlePaddle | Reddit #### Text Classification 模型名称 | 框架 | 数据集 -------- | ------ | ---- [GCN](gnn/text_classification/GCN/mindspore/README.md) | MindSpore | CORA & Citeseer [GCN](gnn/text_classification/GCN/paddlepaddle/README.md) | PaddlePaddle | CORA & PubMed & Citeseer ### HPC #### Molecular Dynamics 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Water/se_e2_a](hpc/molecular_dynamics/water_se_e2_a/tensorflow/README.md) | TensorFlow (DeePMD-kit) | data_water ### Multimodal 模型名称 | 框架 | 数据集 -------- | ------ | ---- [BLIP](multimodal/BLIP/pytorch/README.md) | PyTorch | COCO [CLIP](multimodal/Language-Image_Pre-Training/clip/pytorch/README.md) | PyTorch | CIFAR100 [ControlNet](multimodal/diffusion/ControlNet/README.md) | PyTorch | Fill50K [DDPM](multimodal/diffusion/ddpm/README.md) | PyTorch | CIFAR-10 [L-Verse](multimodal/Language-Image_Pre-Training/L-Verse/pytorch/README.md) | PyTorch | ImageNet [Stable Diffusion 1.4](multimodal/diffusion/stable-diffusion/training/README.md) | PyTorch | pokemon-images [Stable Diffusion 1.5](multimodal/diffusion/stable-diffusion/sd_1.5/README.md) | PyTorch | pokemon-images [Stable Diffusion 2.1](multimodal/diffusion/stable-diffusion/sd_2.1/README.md) | PyTorch | pokemon-images [Stable Diffusion XL](multimodal/diffusion/stable-diffusion/sd_xl/README.md) | PyTorch | pokemon-images ### NLP #### Cloze Test 模型名称 | 框架 | 数据集 -------- | ------ | ---- [GLM](nlp/cloze_test/glm/pytorch/GLMForMultiTokenCloze/README.md) | PyTorch | GLMForMultiTokenCloze #### Dialogue Generation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [CPM](nlp/dialogue_generation/cpm/pytorch/README.md) | PyTorch | STC #### Language Modeling 模型名称 | 框架 | 数据集 -------- | ------ | ---- [BART](nlp/language_model/bart_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | RTE [BERT NER](nlp/ner/bert/pytorch/README.md) | PyTorch | CoNLL-2003 [BERT Pretraining](nlp/language_model/bert/pytorch/README.md) | PyTorch | MLCommon Wikipedia (2048_shards_uncompressed) [BERT Pretraining](nlp/language_model/bert/paddlepaddle/README.md) | PaddlePaddle | MNLI [BERT Pretraining](nlp/language_model/bert/tensorflow/base/README.md) | TensorFlow | MNLI [BERT Pretraining](nlp/language_model/bert/mindspore/README.md) | MindSpore | SQuAD [BERT Text Classification](nlp/text_classification/bert/pytorch/README.md) |PyTorch | GLUE [BERT Text Summerization](nlp/text_summarisation/bert/pytorch/README.md) | PyTorch | cnn_dailymail [BERT Question Answering](nlp/question_answering/bert/pytorch/README.md) | PyTorch | SQuAD [GPT2-Medium-EN](nlp/llm/GPT/paddlepaddle/README.md) | PaddlePaddle | SST-2 [RoBERTa](nlp/language_model/roberta_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | RTE [XLNet](nlp/language_model/xlnet/paddlepaddle/README.md)  | PaddlePaddle | SST-2 #### Large Language Model (LLM) 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Aquila2-34B](nlp/llm/aquila2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus [Baichuan2-7B](nlp/llm/baichuan2-7b/Baichuan2/README.md) | PyTorch (DeepSpeed) | baichuan2-7b-base [Bloom-7B1](nlp/llm/bloom-7b1/firefly/README.md) | PyTorch (Firefly) | school_math_0.25M & bloom-7b1 [ChatGLM-6B](nlp/llm/chatglm-6b/deepspeed/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm-6b [ChatGLM2-6B SFT](nlp/llm/ChatGLM2-6b-sft/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm2-6b [ChatGLM3-6B](nlp/llm/chatglm3-6b/deepspeed/finetune_demo/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm3-6b [Llama-7B](nlp/llm/llama-7b/colossalai/README.md) | PyTorch (Colossal-AI) | llama-7b-hf [Llama2-7B](nlp/llm/llama2-7b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus [Llama2-7B Reward Model Finetuning](nlp/llm/llama2-7b_reward_sft/deepspeed/README.md) | PyTorch (DeepSpeed) | Dahoas/rm-static [Llama2-7B RLHF](nlp/llm/llama2-7b_rlhf/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | llama2-7b&tiny-llama [Llama2-7B SFT](nlp/llm/llama2-7b_sft/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | gpt_small-117M [Llama2-13B](nlp/llm/llama2-13b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus [Llama2-34B](nlp/llm/llama2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus [Llama3-8B](nlp/llm/llama3-8b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus [QWen-7B](nlp/llm/qwen-7b/firefly/README.md) | PyTorch (Firefly) | qwen-7b [QWen1.5-7B](nlp/llm/qwen1.5-7b/firefly/README.md) | PyTorch (Firefly) | school_math [QWen1.5-14B](nlp/llm/qwen1.5-14b/firefly/README.md) | PyTorch (Firefly) | school_math #### Text Correction 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Ernie](nlp/text_correction/ernie/paddlepaddle/README.md) | PaddlePaddle | corpus #### Translation 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Convolutional](nlp/translation/convolutional_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | WMT14 [T5](nlp/translation/t5/pytorch/README.md) | PyTorch | wmt14-en-de-pre-processed [Transformer](nlp/translation/transformer/paddlepaddle/README.md) | PaddlePaddle | wmt14-en-de-pre-processed [Transformer](nlp/translation/transformer_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | IWSLT14 ### Recommendation #### Collaborative Filtering 模型名称 | 框架 | 数据集 -------- | ------ | ---- [NCF](recommendation/collaborative_filtering/ncf/pytorch/README.md) | PyTorch | movielens #### Click Through Rate 模型名称 | 框架 | 数据集 -------- | ------ | ---- [DLRM](recommendation/ctr/dlrm/pytorch/README.md) | PyTorch | Criteo_Terabyte [DLRM](recommendation/ctr/dlrm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte [FFM](recommendation/ctr/ffm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte [DeepFM](recommendation/ctr/deepfm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte [Wide&Deep](recommendation/ctr/wide_deep/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte [xDeepFM](recommendation/ctr/xdeepfm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte ### Reinforcement Learning 模型名称 | 框架 | 数据集 -------- | ------ | ---- [DQN](reinforcement_learning/DQN/paddlepaddle/README.md) | PaddlePaddle | CartPole-v0 ### Speech #### Speech Recognition 模型名称 | 框架 | 数据集 -------- | ------ | ---- [Conformer](speech/speech_recognition/conformer_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL [Efficient Conformer v2](speech/speech_recognition/efficient_conformer_v2_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL [PP-ASR-Conformer](speech/speech_recognition/conformer/paddlepaddle/README.md) | PaddlePaddle | AISHELL [RNN-T](speech/speech_recognition/rnnt/pytorch/README.md) | PyTorch | LJSpeech [Transformer](speech/speech_recognition/transformer_wenet/pytorch/README.md)  | PyTorch (WeNet) | AISHELL [U2++ Conformer](speech/speech_recognition/u2++_conformer_wenet/pytorch/README.md)  | PyTorch (WeNet) | AISHELL [Unified Conformer](speech/speech_recognition/unified_conformer_wenet/pytorch/README.md)  | PyTorch (WeNet) | AISHELL #### Speech Synthesis 模型名称 | 框架 | 数据集 -------- | ------ | ---- [PP-TTS-FastSpeech2](speech/speech_synthesis/fastspeech2/paddlepaddle/README.md) | PaddlePaddle | CSMSC [PP-TTS-HiFiGAN](speech/speech_synthesis/hifigan/paddlepaddle/README.md) | PaddlePaddle | CSMSC [Tacotron2](speech/speech_synthesis/tacotron2/pytorch/README.md) | PyTorch | LJSpeech [VQMIVC](speech/speech_synthesis/vqmivc/pytorch/README.md) | PyTorch | VCTK-Corpus [WaveGlow](speech/speech_synthesis/waveglow/pytorch/README.md) | PyTorch | LJSpeech ### 3D Reconstruction 模型名称 | 框架 | 数据集 -------- | ------ | ---- [HashNeRF](3d-reconstruction/ngp-nerf/pytorch/README.md) | PyTorch | fox -------- ## 容器镜像构建方式 社区用户可参考[容器镜像构建说明](docker/Iluvatar/README.md)在本地构建出能够运行DeepSparkHub仓库中模型的容器镜像。 -------- ## 社区 ### 治理 请参见 DeepSpark Code of Conduct on [Gitee](https://gitee.com/deep-spark/deepspark/blob/master/CODE_OF_CONDUCT.md) or on [GitHub](https://github.com/Deep-Spark/deepspark/blob/main/CODE_OF_CONDUCT.md)。 ### 交流 请联系 。 ### 贡献 请参见 [DeepSparkHub Contributing Guidelines](CONTRIBUTING.md)。 ## 许可证 本项目许可证遵循[Apache-2.0](LICENSE)。