made changes in score_window
Browse files- inference.py +4 -1
inference.py
CHANGED
|
@@ -91,7 +91,10 @@ class SlidingWindowRefiner:
|
|
| 91 |
roi = cv2.resize(roi, self.template.shape[::-1])
|
| 92 |
kp, desc = self.sift.detectAndCompute(roi, None)
|
| 93 |
|
| 94 |
-
if desc is None:
|
|
|
|
|
|
|
|
|
|
| 95 |
return 0
|
| 96 |
|
| 97 |
matches = self.flann.knnMatch(self.template_desc, desc, k=2)
|
|
|
|
| 91 |
roi = cv2.resize(roi, self.template.shape[::-1])
|
| 92 |
kp, desc = self.sift.detectAndCompute(roi, None)
|
| 93 |
|
| 94 |
+
if desc is None or len(desc) < 2:
|
| 95 |
+
return 0
|
| 96 |
+
|
| 97 |
+
if len(self.template_desc) < 2:
|
| 98 |
return 0
|
| 99 |
|
| 100 |
matches = self.flann.knnMatch(self.template_desc, desc, k=2)
|