未验证 提交 34efe848 编写于 作者: G George Ni 提交者: GitHub

[MOT] remove cython_bbox (#3064)

* remove cython_bbox, test=document_fix

* add cython_bbox install log, test=document_fix
上级 f29d636a
......@@ -21,6 +21,9 @@ import numpy as np
from scipy.spatial.distance import cdist
from ..motion import kalman_filter
from ppdet.utils.logger import setup_logger
logger = setup_logger(__name__)
__all__ = [
'merge_matches',
'linear_assignment',
......@@ -70,7 +73,13 @@ def cython_bbox_ious(atlbrs, btlbrs):
ious = np.zeros((len(atlbrs), len(btlbrs)), dtype=np.float)
if ious.size == 0:
return ious
import cython_bbox
try:
import cython_bbox
except Exception as e:
logger.error('cython_bbox not found, please install cython_bbox.'
'for example: `pip install cython_bbox`.')
raise e
ious = cython_bbox.bbox_overlaps(
np.ascontiguousarray(
atlbrs, dtype=np.float),
......
......@@ -12,6 +12,5 @@ pycocotools
setuptools>=42.0.0
lap
sklearn
cython_bbox
motmetrics
openpyxl
openpyxl
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册