未验证 提交 b20e9023 编写于 作者: W wangguanzhong 提交者: GitHub

Cherry pick add citation (#4485)

* update citation

* update citation
上级 0989caf7
......@@ -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
......
......@@ -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>
......
......@@ -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>
......
......@@ -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>
......
......@@ -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>
......
......@@ -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"
......
......@@ -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
......
......@@ -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
......
......@@ -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
......
......@@ -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)
"""
......
......@@ -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)
......@@ -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
......
......@@ -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.
先完成此消息的编辑!
想要评论请 注册