소스 검색

Fix cast exception when specifying ransac parameters

Steyer, Nick 1 년 전
부모
커밋
aff5e08b91
1개의 변경된 파일6개의 추가작업 그리고 6개의 파일을 삭제
  1. 6 6
      hgcalc.py

+ 6 - 6
hgcalc.py

@@ -447,14 +447,14 @@ def main():
     z = 0.99  # total probability of success after all iterations
 
     if len(sys.argv) == 3:
-        threshold = sys.argv[2]
+        threshold = float(sys.argv[2])
     elif len(sys.argv) == 4:
-        threshold = sys.argv[2]
-        p = sys.argv[3]
+        threshold = float(sys.argv[2])
+        p = float(sys.argv[3])
     elif len(sys.argv) == 5:
-        threshold = sys.argv[2]
-        p = sys.argv[3]
-        z = sys.argv[4]
+        threshold = float(sys.argv[2])
+        p = float(sys.argv[3])
+        z = float(sys.argv[4])
     elif len(sys.argv) != 2:
         print('Usage: hgcalc <pairs_file_path> [threshold [valid_correspondence_probability [total_probability]]]')
         return