Slikce

assignment4/cam.py

@@ -16,7 +16,8 @@ def start_realtime_keypoint_detection():
         ret,frame = cap.read()
         frame = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY) # Convert to grayscale
         if MY_SIFT:
-            sift_keypoints,_  = uz_image.sift(frame)
+            #sift_keypoints,_  = uz_image.hessian(frame)
+            _, sift_keypoints = uz_image.hessian_points(frame, 3, 0.004)
             sift_keypoints = np.float64(sift_keypoints)
             kps = []
             for keypoint in sift_keypoints:
@@ -26,7 +27,8 @@ def start_realtime_keypoint_detection():
             kps, _ = sift.detectAndCompute(frame, None)
         frame = cv2.drawKeypoints(frame, kps, None, flags=cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS)
         #cv2.imshow('hello', frame)
-        cv2.imwrite("./datam/frames/%d.jpg" % count, frame)
+        cv2.imwrite("./datam/hessian/%d.jpg" % count, frame)
+        print(f'{count} <- frame')
         count = count + 1
         #if cv2.waitKey(10) & 0xFF == ord('q'):
             #break