计算机视觉四大任务模型汇总

计算机视觉中有四大核心任务:

1-分类任务、2-目标检测任务、3-目标分割任务 和 4-关键点检测任务

文章1:

一文读懂计算机视觉4大任务

文章2:

图像的目标分割任务:语义分割和实例分割

不同任务之间相关但不完全相同,因此不同的任务最好选择相应的模型,话不多说,看表:

注:表中github链接并不一定是模型的正式版本,只是本文用于展示模型的网络结构和应用

1-分类任务模型

|--------|------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 序号 | 模型 | ipynb模型的github链接 |
| 1 | LeNet | https://github.com/udacity/CarND-LeNet-Lab |
| 2 | AlexNet | https://github.com/Fannjh/AlexNet-TF |
| 3 | VGGNet | https://github.com/Fozan-Talat/Image-Classifier-VGG |
| 4 | GoogLeNet | GitHub - AbdelrahmanShehata482/CNN-project: CNN_Project (py and ipynb code ) (Vgg16-GoogleNet from scratch) |
| 5 | ResNet | GitHub - ry/tensorflow-resnet: ResNet model in TensorFlow |
| 6 | DenseNet | GitHub - titu1994/DenseNet: DenseNet implementation in Keras |
| 7 | MobileNet | https://github.com/Zehaos/MobileNet |
| 8 | EfficientNet | https://github.com/qubvel/efficientnet |
| 9 | SVM(支持向量机) | https://github.com/Think103/- |

2-目标检测任务模型

|--------|---------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 序号 | 模型 | ipynb模型的github链接 |
| 1 | R-CNN(已过时) | |
| 2 | Fast R-CNN(已过时) | |
| 3 | Faster R-CNN | GitHub - kbardool/Keras-frcnn: Keras Implementation of Faster R-CNN |
| 4 | YOLO | https://github.com/ultralytics/yolov5 |
| 5 | SSD | https://github.com/lufficc/SSD |
| 6 | RetinaNet | https://github.com/fizyr/keras-retinanet |
| 7 | Mask R-CNN | https://github.com/SanmathiK/PedNet |
| 8 | EfficientDet | GitHub - xuannianz/EfficientDet: EfficientDet (Scalable and Efficient Object Detection) implementation in Keras and Tensorflow |
| 9 | CenterNet | https://github.com/xingyizhou/CenterNet |

3-目标分割任务模型

|--------|----------|----------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 序号 | 分割类型 | 模型 | ipynb模型的github链接 |
| 1 | 语义分割 | FCN | GitHub - wkentaro/pytorch-fcn: PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.) |
| 2 | 语义分割 | U-Net | GitHub - yingkaisha/keras-unet-collection: The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones. |
| 3 | 语义分割 | DeepLab | GitHub - fregu856/deeplabv3: PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset. |
| 4 | 语义分割 | PSPNet | GitHub - Lextal/pspnet-pytorch: PyTorch implementation of PSPNet segmentation network |
| 5 | 语义分割 | SegNet | GitHub - preddy5/segnet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation |
| 6 | 语义分割 | HRNet | GitHub - HRNet/HRNet-Image-Classification: Train the HRNet model on ImageNet |
| 7 | 实例分割 | Mask R-CNN | https://github.com/saikoneru/Instance-Segementation |
| 8 | 实例分割 | PANet | https://github.com/kaixin96/PANet |
| 9 | 实例分割 | YOLACT | https://github.com/dbolya/yolact |
| 10 | 实例分割 | SOLO | https://github.com/iambankaratharva/SOLO-Instance-Segmentation |
| 11 | 实例分割 | PointRend | https://github.com/zsef123/PointRend-PyTorch |

4-关键点检测任务模型

|--------|----------|-----------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 序号 | 检测目标 | 模型 | ipynb模型的github链接 |
| 1 | 人脸 | Dlib | GitHub - davisking/dlib: A toolkit for making real world machine learning and data analysis applications in C++ |
| 2 | 人脸 | MTCNN | GitHub - ipazc/mtcnn: MTCNN face detection implementation for TensorFlow, as a PIP package. |
| 3 | 人脸 | FaceBoxes | GitHub - zisianw/FaceBoxes.PyTorch: A PyTorch Implementation of FaceBoxes |
| 4 | 人脸 | PRNet | GitHub - yfeng95/PRNet: Joint 3D Face Reconstruction and Dense Alignment with Position Map Regression Network (ECCV 2018) |
| 5 | 人体 | OpenPose | GitHub - Hzzone/pytorch-openpose: pytorch implementation of openpose including Hand and Body Pose Estimation. |
| 6 | 人体 | HRNet | https://github.com/HRNet/HigherHRNet-Human-Pose-Estimation |
| 7 | 人体 | CPM | GitHub - PanZiqiAI/CPM-Clothes-Keypoints-Detection: Convolutional Pose Machine implemented for clothes key points detection. |
| 8 | 人体 | Mask R-CNN with Keypoint Detection Branch | GitHub - chrispolo/Keypoints-of-humanpose-with-Mask-R-CNN: Use the Mask RCNN for the human pose estimation |
| 9 | 人体 | AlphaPose | GitHub - Amanbhandula/AlphaPose: AlphaPose Implementation in Pytorch along with the pre-trained weights |
| 10 | 人体 | MoveNet | GitHub - fire717/movenet.pytorch: A Pytorch implementation of MoveNet from Google. Include training code and pre-trained model. |

需要说明,上表中模型名称并不单指某个模型,而是一类模型统称,如YOLO模型实际包括了yolov1~yolov10的10个系列。

相关推荐
OpenVINO 中文社区2 分钟前
OpenVINO™正式进入 llama.cpp:GGUF 模型现已支持 Intel CPU、GPU 与 NPU
人工智能·openvino·英特尔
ZKNOW甄知科技7 分钟前
数智同行:甄知科技2026年Q1季度回顾
运维·服务器·人工智能·科技·程序人生·安全·自动化
呆呆敲代码的小Y8 分钟前
【Unity工具篇】| 游戏完整资源热更新流程,YooAsset官方示例项目
人工智能·游戏·unity·游戏引擎·热更新·yooasset·免费游戏
jikemaoshiyanshi8 分钟前
B2B企业GEO服务商哪家好?深度解析径硕科技(JINGdigital)及其JINGEO产品为何是首选
大数据·运维·人工智能·科技
Lab_AI8 分钟前
浩天药业携手创腾科技,开启研发数字化新篇章!电子实验记录本(ELN)落地浩天药业
人工智能
supericeice10 分钟前
大模型建筑隐患管理方案怎么做?创邻科技用知识图谱、图数据库和企业AI大脑打通隐患问答、整改与推荐
人工智能·科技·知识图谱
蕤葳-14 分钟前
非编程背景学习AI的方法
人工智能
北京耐用通信17 分钟前
不换设备、不重写程序:耐达讯自动化网关如何实现CC-Link IE转Modbus TCP的高效互通?
人工智能·科技·物联网·网络协议·自动化·信息与通信
计算机毕业设计指导17 分钟前
基于机器学习和深度学习的恶意WebURL检测系统实战详解
人工智能·深度学习·机器学习·网络安全
珂朵莉MM19 分钟前
第七届全球校园人工智能算法精英大赛-算法巅峰赛产业命题赛第3赛季优化题--多策略混合算法
人工智能·算法