栏目分类:
子分类:
返回
终身学习网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
终身学习网 > IT > 软件开发 > 后端开发 > Python

PaddleDetection代码学习(一)综述

Python 更新时间:发布时间: 百科书网 趣学号

2021SC@SDUSC

PaddleDetection是飞桨推出的端到端目标检测开发套件,旨在帮助开发者更快更好地完成检测模型的训练、精度速度优化到部署全流程。PaddleDetection以模块化的设计实现了多种主流目标检测算法,并且提供了丰富的数据增强、网络组件、损失函数等模块,集成了模型压缩和跨平台高性能部署能力。目前基于PaddleDetection已经完成落地的项目涉及工业质检、遥感图像检测、无人巡检等多个领域。

PaddleDetection安装: 创建python环境

输入命令:

conda create -n paddle python=3.6

激活环境: 

activate paddle

安装gpu框架
1、首先看自己的CUDA、Cudnn的版本是多少,我的CUDA版本是9.0,Cudnn是7.4
如果是CUDA=10.0 则下方语句中改成 .post107
2、如果不使用镜像的话下载速度会比较慢,加上镜像加快速度
3、现在大多数paddledetection的api都在1.6以上,所以这里下载gpu1.8版本的, .post97分别对应的CUDA、Cudnn的版本号码
输入命令:

python -m pip install paddlepaddle-gpu==1.8.1.post97 -i https://mirror.baidu.com/pypi/simple

测试环境:
进入python环境分别输入以下语句: 

import paddle.fluid as fluid

fluid.install_check.run_check()

输出:

安装gpu框架完成
安装coco-api:
输入命令:

pip install Cython

# 我下载时遇到了这个问题,需要更新pip,更新后重新下载
python -m pip install --upgrade pip

下载api:

pip install git+https://github.com/philferriere/cocoapi.git

最终完成后:

 

下载PaddleDetection

输入命令:(路径下)

pip install -r requirements.txt

 也可以直接在github上下载:GitHub - PaddlePaddle/PaddleDetection: Object detection and instance segmentation toolkit based on PaddlePaddle.
安装Python依赖库:

pip install -r requirements.txt

 测试是否安装成功:

python ppdet/modeling/tests/test_architectures.py

小组分工:

我主要负责基础框架代码分析和Anchor Free系列模型

Anchor Free系列模型

简介:目前主流的检测算法大体分为两类: single-stage和two-stage,其中single-stage的经典算法包括SSD, YOLO等,two-stage方法有RCNN系列模型,两大类算法在PaddleDetection Model Zoo中均有给出,它们的共同特点是先定义一系列密集的,大小不等的anchor区域,再基于这些先验区域进行分类和回归,这种方式极大的受限于anchor自身的设计。随着CornerNet的提出,涌现了多种anchor free方法,PaddleDetection也集成了一系列anchor free算法。

算法细节 CornerNet-Squeeze

简介: CornerNet-Squeeze 在Cornernet基础上进行改进,预测目标框的左上角和右下角的位置,同时参考SqueezeNet和MobileNet的特点,优化了CornerNet骨干网络Hourglass-104,大幅提升了模型预测速度,相较于原版YOLO-v3,在训练精度和推理速度上都具备一定优势。

FCOS

简介: FCOS是一种密集预测的anchor-free检测算法,使用RetinaNet的骨架,直接在feature map上回归目标物体的长宽,并预测物体的类别以及centerness(feature map上像素点离物体中心的偏移程度),centerness最终会作为权重来调整物体得分。

TTFNet

简介: TTFNet是一种用于实时目标检测且对训练时间友好的网络,对CenterNet收敛速度慢的问题进行改进,提出了利用高斯核生成训练样本的新方法,有效的消除了anchor-free head中存在的模糊性。同时简单轻量化的网络结构也易于进行任务扩展。

本文部分引用:https://blog.csdn.net/qq_43656233/article/details/106248013 

 

 


 

转载请注明:文章转载自 www.051e.com
本文地址:http://www.051e.com/it/280566.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号