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

update citation (#4464)

* update citation

* update citation
上级 a229530b
...@@ -12,6 +12,10 @@ ...@@ -12,6 +12,10 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 #ifndef LAPJV_H
#define LAPJV_H #define LAPJV_H
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 #pragma once
#include <map> #include <map>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 #pragma once
#include <vector> #include <vector>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 <map>
#include <stdio.h> #include <stdio.h>
#include <limits.h> #include <limits.h>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 <algorithm>
#include "include/trajectory.h" #include "include/trajectory.h"
......
...@@ -12,6 +12,10 @@ ...@@ -12,6 +12,10 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 #ifndef LAPJV_H
#define LAPJV_H #define LAPJV_H
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 #pragma once
#include <map> #include <map>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 #pragma once
#include <vector> #include <vector>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 <map>
#include <stdio.h> #include <stdio.h>
#include <limits.h> #include <limits.h>
......
...@@ -12,6 +12,11 @@ ...@@ -12,6 +12,11 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // 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 <algorithm>
#include "include/trajectory.h" #include "include/trajectory.h"
......
...@@ -12,6 +12,9 @@ ...@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # 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 absolute_import
from __future__ import print_function from __future__ import print_function
from __future__ import division from __future__ import division
......
...@@ -141,6 +141,13 @@ class BBoxPostProcess(nn.Layer): ...@@ -141,6 +141,13 @@ class BBoxPostProcess(nn.Layer):
@register @register
class MaskPostProcess(object): 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): def __init__(self, binary_thresh=0.5):
super(MaskPostProcess, self).__init__() super(MaskPostProcess, self).__init__()
self.binary_thresh = binary_thresh self.binary_thresh = binary_thresh
......
...@@ -12,6 +12,9 @@ ...@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# The code is based on
# https://github.com/facebookresearch/detectron2/blob/main/detectron2/modeling/anchor_generator.py
import math import math
import paddle import paddle
......
...@@ -279,9 +279,12 @@ def sample_bbox(matches, ...@@ -279,9 +279,12 @@ def sample_bbox(matches,
def polygons_to_mask(polygons, height, width): def polygons_to_mask(polygons, height, width):
""" """
Convert the polygons to mask format
Args: Args:
polygons (list[ndarray]): each array has shape (Nx2,) polygons (list[ndarray]): each array has shape (Nx2,)
height, width (int) height (int): mask height
width (int): mask width
Returns: Returns:
ndarray: a bool mask of shape (height, width) ndarray: a bool mask of shape (height, width)
""" """
......
...@@ -12,22 +12,14 @@ ...@@ -12,22 +12,14 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # 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 from collections import namedtuple
class ShapeSpec( class ShapeSpec(
namedtuple("_ShapeSpec", ["channels", "height", "width", "stride"])): 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): def __new__(cls, channels=None, height=None, width=None, stride=None):
return super(ShapeSpec, cls).__new__(cls, channels, height, width, return super(ShapeSpec, cls).__new__(cls, channels, height, width,
stride) stride)
...@@ -12,6 +12,9 @@ ...@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # 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 absolute_import
from __future__ import print_function from __future__ import print_function
from __future__ import division from __future__ import division
......
...@@ -35,7 +35,12 @@ def box_flip(boxes, im_shape): ...@@ -35,7 +35,12 @@ def box_flip(boxes, im_shape):
def nms(dets, thresh): 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: if dets.shape[0] == 0:
return dets[[], :] return dets[[], :]
scores = dets[:, 0] scores = dets[:, 0]
...@@ -91,6 +96,10 @@ def nms(dets, thresh): ...@@ -91,6 +96,10 @@ def nms(dets, thresh):
def soft_nms(dets, sigma, thres): def soft_nms(dets, sigma, thres):
"""
refer to:
https://github.com/facebookresearch/Detectron/blob/main/detectron/utils/cython_nms.pyx
"""
dets_final = [] dets_final = []
while len(dets) > 0: while len(dets) > 0:
maxpos = np.argmax(dets[:, 0]) maxpos = np.argmax(dets[:, 0])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册