Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
48ddaa99
P
PaddleDetection
项目概览
s920243400
/
PaddleDetection
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleDetection
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
48ddaa99
编写于
12月 18, 2020
作者:
Q
qingqing01
提交者:
GitHub
12月 18, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update docs (#1926)
上级
bc563833
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
32 addition
and
26 deletion
+32
-26
README.md
README.md
+5
-5
docs/tutorials/INSTALL_cn.md
docs/tutorials/INSTALL_cn.md
+1
-1
docs/tutorials/PrepareDataSet.md
docs/tutorials/PrepareDataSet.md
+10
-4
ppdet/modeling/bbox.py
ppdet/modeling/bbox.py
+5
-16
ppdet/utils/checkpoint.py
ppdet/utils/checkpoint.py
+11
-0
未找到文件。
README.md
浏览文件 @
48ddaa99
# PaddleDetection
动态图版本的PaddleDetection, 支持的模型:
动态图版本的PaddleDetection, 此版本为试用版本,还在持续优化设计、性能、新增模型、文档等。
支持的模型:
-
Faster-RCNN (FPN)
-
Mask-RCNN (FPN)
-
Cascade RCNN
-
YOLOv3
-
SSD
扩展特性:
...
...
@@ -19,7 +23,6 @@
### 教程
-
[
安装说明
](
docs/tutorials/INSTALL_cn.md
)
-
[
快速开始
](
docs/tutorials/QUICK_STARTED_cn.md
)
-
[
训练/评估/预测流程
](
docs/tutorials/GETTING_STARTED_cn.md
)
-
[
常见问题汇总
](
docs/FAQ.md
)
-
[
推理部署
](
deploy
)
...
...
@@ -30,6 +33,3 @@
## 模型库
-
[
模型库
](
docs/MODEL_ZOO_cn.md
)
## 许可证书
本项目的发布受
[
Apache 2.0 license
](
LICENSE
)
许可认证。
docs/tutorials/INSTALL_cn.md
浏览文件 @
48ddaa99
...
...
@@ -69,6 +69,6 @@ git clone https://github.com/PaddlePaddle/PaddleDetection.git
也可以通过
[
https://gitee.com/paddlepaddle/PaddleDetection
](
https://gitee.com/paddlepaddle/PaddleDetection
)
克隆。
```
cd <path/to/clone/Pa
zccccccccccc
ddleDetection>
cd <path/to/clone/PaddleDetection>
git clone https://gitee.com/paddlepaddle/PaddleDetection
```
docs/tutorials/PrepareDataSet.md
浏览文件 @
48ddaa99
...
...
@@ -15,6 +15,7 @@
-
[
用户数据数据转换示例
](
#用户数据数据转换示例
)
### 目标检测数据说明
目标检测的数据比分类复杂,一张图像中,需要标记出各个目标区域的位置和类别。
一般的目标区域位置用一个矩形框来表示,一般用以下3种方式表达:
...
...
@@ -28,6 +29,7 @@
常见的目标检测数据集如Pascal VOC和COCO,采用的是第一种
`x1,y1,x2,y2`
表示物体的bounding box.
### 准备训练数据
PaddleDetection默认支持
[
COCO
](
http://cocodataset.org
)
和
[
Pascal VOC
](
http://host.robots.ox.ac.uk/pascal/VOC/
)
和
[
WIDER-FACE
](
http://shuoyang1213.me/WIDERFACE/
)
数据源。
同时还支持自定义数据源,包括:
...
...
@@ -43,8 +45,10 @@ ppdet_root=$(pwd)
```
#### VOC数据数据
VOC数据是
[
Pascal VOC
](
http://host.robots.ox.ac.uk/pascal/VOC/
)
比赛使用的数据。Pascal VOC比赛不仅包含图像分类分类任务,还包含图像目标检测、图像分割等任务,其标注文件中包含多个任务的标注内容。
VOC数据集指的是Pascal VOC比赛使用的数据。用户自定义的VOC数据,xml文件中的非必须字段,请根据实际情况选择是否标注或是否使用默认值。
##### VOC数据集下载
-
通过代码自动化下载VOC数据集
...
...
@@ -114,6 +118,7 @@ VOC数据集指的是Pascal VOC比赛使用的数据。用户自定义的VOC数
按照如上数据文件组织结构组织文件即可。
##### VOC数据标注文件介绍
VOC数据是每个图像文件对应一个同名的xml文件,xml文件中标记物体框的坐标和类别等信息。例如图像
`2007_002055.jpg`
:
![](
../images/2007_002055.jpg
)
...
...
@@ -410,8 +415,9 @@ roadsign数据集统计:
| 数据 | 图片数量 |
| :--------: | :-----------: |
| train | 701 |
| valid | 176 |
| train | 701 |
| valid | 176 |
**说明:**
(1)用户数据,建议在训练前仔细检查数据,避免因数据标注格式错误或图像数据不完整造成训练过程中的crash
(2)如果图像尺寸太大的话,在不限制读入数据尺寸情况下,占用内存较多,会造成内存/显存溢出,请合理设置batch_size,可从小到大尝试
**说明:**
(1)用户数据,建议在训练前仔细检查数据,避免因数据标注格式错误或图像数据不完整造成训练过程中的crash
(2)如果图像尺寸太大的话,在不限制读入数据尺寸情况下,占用内存较多,会造成内存/显存溢出,请合理设置batch_size,可从小到大尝试
ppdet/modeling/bbox.py
浏览文件 @
48ddaa99
...
...
@@ -45,10 +45,11 @@ class Anchor(object):
return
rpn_scores
,
rpn_deltas
,
anchors
def
generate_loss_inputs
(
self
,
inputs
,
rpn_head_out
,
anchors
):
assert
len
(
rpn_head_out
)
==
len
(
anchors
),
"rpn_head_out and anchors should have same length, but received rpn_head_out' length is {} and anchors' length is {}"
.
format
(
len
(
rpn_head_out
),
len
(
anchors
))
if
len
(
rpn_head_out
)
!=
len
(
anchors
):
raise
ValueError
(
"rpn_head_out and anchors should have same length, "
" but received rpn_head_out' length is {} and anchors' "
" length is {}"
.
format
(
len
(
rpn_head_out
),
len
(
anchors
)))
rpn_score
,
rpn_delta
,
anchors
=
self
.
_get_target_input
(
rpn_head_out
,
anchors
)
...
...
@@ -69,18 +70,6 @@ class Anchor(object):
return
outs
@
register
class
AnchorYOLO
(
object
):
__inject__
=
[
'anchor_generator'
]
def
__init__
(
self
,
anchor_generator
):
super
(
AnchorYOLO
,
self
).
__init__
()
self
.
anchor_generator
=
anchor_generator
def
__call__
(
self
):
return
self
.
anchor_generator
()
@
register
class
Proposal
(
object
):
__inject__
=
[
'proposal_generator'
,
'proposal_target_generator'
]
...
...
ppdet/utils/checkpoint.py
浏览文件 @
48ddaa99
...
...
@@ -147,6 +147,17 @@ def load_pretrain_weight(model,
def
save_model
(
model
,
optimizer
,
save_dir
,
save_name
,
last_epoch
):
"""
save model into disk.
Args:
model (paddle.nn.Layer): the Layer instalce to save parameters.
optimizer (paddle.optimizer.Optimizer): the Optimizer instance to
save optimizer states.
save_dir (str): the directory to be saved.
save_name (str): the path to be saved.
last_epoch (int): the epoch index.
"""
if
not
os
.
path
.
exists
(
save_dir
):
os
.
makedirs
(
save_dir
)
save_path
=
os
.
path
.
join
(
save_dir
,
save_name
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录