Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
b20e9023
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看板
未验证
提交
b20e9023
编写于
11月 05, 2021
作者:
W
wangguanzhong
提交者:
GitHub
11月 05, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Cherry pick add citation (#4485)
* update citation * update citation
上级
0989caf7
变更
13
隐藏空白更改
内联
并排
Showing
13 changed file
with
62 addition
and
13 deletion
+62
-13
deploy/cpp/include/lapjv.h
deploy/cpp/include/lapjv.h
+4
-0
deploy/cpp/include/tracker.h
deploy/cpp/include/tracker.h
+5
-0
deploy/cpp/include/trajectory.h
deploy/cpp/include/trajectory.h
+5
-0
deploy/cpp/src/lapjv.cpp
deploy/cpp/src/lapjv.cpp
+5
-0
deploy/cpp/src/tracker.cc
deploy/cpp/src/tracker.cc
+5
-0
deploy/cpp/src/trajectory.cc
deploy/cpp/src/trajectory.cc
+5
-0
ppdet/data/transform/gridmask_utils.py
ppdet/data/transform/gridmask_utils.py
+3
-0
ppdet/modeling/post_process.py
ppdet/modeling/post_process.py
+7
-0
ppdet/modeling/proposal_generator/anchor_generator.py
ppdet/modeling/proposal_generator/anchor_generator.py
+3
-0
ppdet/modeling/proposal_generator/target.py
ppdet/modeling/proposal_generator/target.py
+4
-1
ppdet/modeling/shape_spec.py
ppdet/modeling/shape_spec.py
+3
-11
static/ppdet/data/transform/gridmask_utils.py
static/ppdet/data/transform/gridmask_utils.py
+3
-0
static/ppdet/utils/post_process.py
static/ppdet/utils/post_process.py
+10
-1
未找到文件。
deploy/cpp/include/lapjv.h
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,10 @@
// See the License for the specific language governing permissions and
// limitations under the License.
// The code is based on:
// https://github.com/gatagat/lap/blob/master/lap/lapjv.h
// Ths copyright of gatagat/lap is as follows:
// MIT License
#ifndef LAPJV_H
#define LAPJV_H
...
...
deploy/cpp/include/tracker.h
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and
// limitations under the License.
// The code is based on:
// https://github.com/CnybTseng/JDE/blob/master/platforms/common/jdetracker.h
// Ths copyright of CnybTseng/JDE is as follows:
// MIT License
#pragma once
#include <map>
...
...
deploy/cpp/include/trajectory.h
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and
// limitations under the License.
// The code is based on:
// https://github.com/CnybTseng/JDE/blob/master/platforms/common/trajectory.h
// Ths copyright of CnybTseng/JDE is as follows:
// MIT License
#pragma once
#include <vector>
...
...
deploy/cpp/src/lapjv.cpp
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and
// limitations under the License.
// The code is based on:
// https://github.com/gatagat/lap/blob/master/lap/lapjv.cpp
// Ths copyright of gatagat/lap is as follows:
// MIT License
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
...
...
deploy/cpp/src/tracker.cc
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and
// limitations under the License.
// The code is based on:
// https://github.com/CnybTseng/JDE/blob/master/platforms/common/jdetracker.cpp
// Ths copyright of CnybTseng/JDE is as follows:
// MIT License
#include <map>
#include <stdio.h>
#include <limits.h>
...
...
deploy/cpp/src/trajectory.cc
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and
// limitations under the License.
// The code is based on:
// https://github.com/CnybTseng/JDE/blob/master/platforms/common/trajectory.cpp
// Ths copyright of CnybTseng/JDE is as follows:
// MIT License
#include <algorithm>
#include "include/trajectory.h"
...
...
ppdet/data/transform/gridmask_utils.py
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# The code is based on:
# https://github.com/dvlab-research/GridMask/blob/master/detection_grid/maskrcnn_benchmark/data/transforms/grid.py
from
__future__
import
absolute_import
from
__future__
import
print_function
from
__future__
import
division
...
...
ppdet/modeling/post_process.py
浏览文件 @
b20e9023
...
...
@@ -141,6 +141,13 @@ class BBoxPostProcess(nn.Layer):
@
register
class
MaskPostProcess
(
object
):
"""
refer to:
https://github.com/facebookresearch/detectron2/layers/mask_ops.py
Get Mask output according to the output from model
"""
def
__init__
(
self
,
binary_thresh
=
0.5
):
super
(
MaskPostProcess
,
self
).
__init__
()
self
.
binary_thresh
=
binary_thresh
...
...
ppdet/modeling/proposal_generator/anchor_generator.py
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# The code is based on
# https://github.com/facebookresearch/detectron2/blob/main/detectron2/modeling/anchor_generator.py
import
math
import
paddle
...
...
ppdet/modeling/proposal_generator/target.py
浏览文件 @
b20e9023
...
...
@@ -279,9 +279,12 @@ def sample_bbox(matches,
def
polygons_to_mask
(
polygons
,
height
,
width
):
"""
Convert the polygons to mask format
Args:
polygons (list[ndarray]): each array has shape (Nx2,)
height, width (int)
height (int): mask height
width (int): mask width
Returns:
ndarray: a bool mask of shape (height, width)
"""
...
...
ppdet/modeling/shape_spec.py
浏览文件 @
b20e9023
...
...
@@ -12,22 +12,14 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# The code is based on:
# https://github.com/facebookresearch/detectron2/blob/main/detectron2/layers/shape_spec.py
from
collections
import
namedtuple
class
ShapeSpec
(
namedtuple
(
"_ShapeSpec"
,
[
"channels"
,
"height"
,
"width"
,
"stride"
])):
"""
A simple structure that contains basic shape specification about a tensor.
It is often used as the auxiliary inputs/outputs of models,
to complement the lack of shape inference ability among paddle modules.
Attributes:
channels:
height:
width:
stride:
"""
def
__new__
(
cls
,
channels
=
None
,
height
=
None
,
width
=
None
,
stride
=
None
):
return
super
(
ShapeSpec
,
cls
).
__new__
(
cls
,
channels
,
height
,
width
,
stride
)
static/ppdet/data/transform/gridmask_utils.py
浏览文件 @
b20e9023
...
...
@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# The code is based on:
# https://github.com/dvlab-research/GridMask/blob/master/detection_grid/maskrcnn_benchmark/data/transforms/grid.py
from
__future__
import
absolute_import
from
__future__
import
print_function
from
__future__
import
division
...
...
static/ppdet/utils/post_process.py
浏览文件 @
b20e9023
...
...
@@ -35,7 +35,12 @@ def box_flip(boxes, im_shape):
def
nms
(
dets
,
thresh
):
"""Apply classic DPM-style greedy NMS."""
"""
refer to:
https://github.com/facebookresearch/Detectron/blob/main/detectron/utils/cython_nms.pyx
Apply classic DPM-style greedy NMS.
"""
if
dets
.
shape
[
0
]
==
0
:
return
dets
[[],
:]
scores
=
dets
[:,
0
]
...
...
@@ -91,6 +96,10 @@ def nms(dets, thresh):
def
soft_nms
(
dets
,
sigma
,
thres
):
"""
refer to:
https://github.com/facebookresearch/Detectron/blob/main/detectron/utils/cython_nms.pyx
"""
dets_final
=
[]
while
len
(
dets
)
>
0
:
maxpos
=
np
.
argmax
(
dets
[:,
0
])
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录