ayushsaun commited on
Commit
e58f0d9
·
1 Parent(s): ebce887

made changes in score_window

Browse files
Files changed (1) hide show
  1. 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)