Skip to content

Commit 07fc848

Browse files
committed
Fix import in test
1 parent 3d8cada commit 07fc848

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

test.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
Make sure the active directory is the directory of the repo when running the test in a IDE.
33
"""
44
from skimage.data import coins
5-
import MTM
6-
from MTM.Detection import plotDetections
7-
print( MTM.__version__ )
5+
import mtm
6+
from mtm.detection import plotDetections
7+
print( mtm.__version__ )
88
import numpy as np
99

1010
#%% Get image and templates by cropping
@@ -20,13 +20,13 @@
2020

2121

2222
#%% Perform matching
23-
listHit = MTM.findMatches(image, listTemplates, listLabels)
24-
singleObject = MTM.findMatches(image, listTemplates, listLabels, nObjects=1) # there should be 1 top hit per template
23+
listHit = mtm.findMatches(image, listTemplates, listLabels)
24+
singleObject = mtm.findMatches(image, listTemplates, listLabels, nObjects=1) # there should be 1 top hit per template
2525

26-
finalHits = MTM.matchTemplates(image,
26+
finalHits = mtm.matchTemplates(image,
2727
listTemplates,
2828
listLabels,
29-
score_threshold=0.4,
29+
scoreThreshold=0.4,
3030
maxOverlap=0)
3131

3232
print("Found {} coins".format(len(finalHits)))

0 commit comments

Comments
 (0)