未验证 提交 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 ...@@ -21,6 +21,9 @@ import numpy as np
from scipy.spatial.distance import cdist from scipy.spatial.distance import cdist
from ..motion import kalman_filter from ..motion import kalman_filter
from ppdet.utils.logger import setup_logger
logger = setup_logger(__name__)
__all__ = [ __all__ = [
'merge_matches', 'merge_matches',
'linear_assignment', 'linear_assignment',
...@@ -70,7 +73,13 @@ def cython_bbox_ious(atlbrs, btlbrs): ...@@ -70,7 +73,13 @@ def cython_bbox_ious(atlbrs, btlbrs):
ious = np.zeros((len(atlbrs), len(btlbrs)), dtype=np.float) ious = np.zeros((len(atlbrs), len(btlbrs)), dtype=np.float)
if ious.size == 0: if ious.size == 0:
return ious 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( ious = cython_bbox.bbox_overlaps(
np.ascontiguousarray( np.ascontiguousarray(
atlbrs, dtype=np.float), atlbrs, dtype=np.float),
......
...@@ -12,6 +12,5 @@ pycocotools ...@@ -12,6 +12,5 @@ pycocotools
setuptools>=42.0.0 setuptools>=42.0.0
lap lap
sklearn sklearn
cython_bbox
motmetrics 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.
先完成此消息的编辑!
想要评论请 注册