Skip to content

Commit 70ad978

Browse files
committed
Merge branch 'fix-new-opencv'
2 parents 8156852 + b52e37d commit 70ad978

File tree

3 files changed

+4
-6
lines changed

3 files changed

+4
-6
lines changed

MTM/NMS.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,8 @@ def NMS(tableHit, scoreThreshold=0.5, sortAscending=False, N_object=float("inf")
7171
indexes = cv2.dnn.NMSBoxes(listBoxes, listScores, scoreThreshold, maxOverlap)
7272

7373
# Get N best hit
74-
if N_object == float("inf"):
75-
indexes = [ index[0] for index in indexes ] # ordered by score
76-
else:
77-
indexes = [ index[0] for index in indexes[:N_object] ]
74+
if N_object != float("inf"):
75+
indexes = indexes[:N_object]
7876

7977
outTable = tableHit.iloc[indexes]
8078

MTM/version.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
# 1) we don't load dependencies by storing it in __init__.py
33
# 2) we can import it in setup.py for the same reason
44
# 3) we can import it into your module module
5-
__version__ = '1.6.2.post1'
5+
__version__ = '1.6.3'

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
packages=["MTM"],
2121
install_requires=[
2222
'numpy',
23-
'opencv-python-headless<=4.3.0',
23+
'opencv-python-headless>=4.5.4',
2424
'scikit-image',
2525
'scipy',
2626
'pandas'

0 commit comments

Comments
 (0)