Update plot.py

plot.py

@@ -62,7 +62,7 @@ def extract_points_from_xml(xml_file):
     return np.array([points[p] for p in order], dtype=np.float32)
 
 # ========= Draw correspondences and (optional) boxes =========
-def draw_feature_matching(img1, pts1, img2, pts2, draw_boxes=True):
+def draw_feature_matching(img1, pts1, img2, pts2, color,draw_boxes=True):
     """
     Draws feature correspondences between two images, handling different sizes.
     """
@@ -90,8 +90,8 @@ def draw_feature_matching(img1, pts1, img2, pts2, draw_boxes=True):
 
     # Optional: Draw polygons (boxes)
     if draw_boxes:
-        cv2.polylines(new_img, [pts1_scaled.reshape((-1,1,2))], True, (0,255,0), 3)
-        cv2.polylines(new_img, [pts2_scaled.reshape((-1,1,2)) + np.array([w1,0])], True, (0,255,0), 3)
+        cv2.polylines(new_img, [pts1_scaled.reshape((-1,1,2))], True, color, 3)
+        cv2.polylines(new_img, [pts2_scaled.reshape((-1,1,2)) + np.array([w1_new,0])], True, color, 3)
 
     # Draw correspondences
     for (x1, y1), (x2, y2) in zip(pts1_scaled, pts2_scaled):