Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -87,18 +87,13 @@ def homography_all_detectors(flat_file, persp_file, json_file, xml_file):
|
|
| 87 |
xml_mapped = cv2.perspectiveTransform(xml_points.reshape(-1,1,2),H).reshape(-1,2)
|
| 88 |
for px,py in xml_mapped: cv2.circle(xml_gt_img,(int(px),int(py)),5,(0,0,255),-1)
|
| 89 |
|
| 90 |
-
# Convert to RGB
|
| 91 |
-
size=(256,256)
|
| 92 |
flat_rgb = cv2.cvtColor(flat_img,cv2.COLOR_BGR2RGB)
|
| 93 |
-
flat_rgb = cv2.resize(flat_rgb,size)
|
| 94 |
match_rgb = cv2.cvtColor(match_img,cv2.COLOR_BGR2RGB)
|
| 95 |
-
match_rgb = cv2.resize(match_rgb,size)
|
| 96 |
roi_rgb = cv2.cvtColor(persp_roi,cv2.COLOR_BGR2RGB)
|
| 97 |
-
roi_rgb = cv2.resize(roi_rgb,size)
|
| 98 |
xml_rgb = cv2.cvtColor(xml_gt_img,cv2.COLOR_BGR2RGB)
|
| 99 |
-
xml_rgb = cv2.resize(xml_rgb,size)
|
| 100 |
|
| 101 |
-
# Merge 2x2 grid
|
| 102 |
top = np.hstack([flat_rgb, match_rgb])
|
| 103 |
bottom = np.hstack([roi_rgb, xml_rgb])
|
| 104 |
combined_grid = np.vstack([top,bottom])
|
|
|
|
| 87 |
xml_mapped = cv2.perspectiveTransform(xml_points.reshape(-1,1,2),H).reshape(-1,2)
|
| 88 |
for px,py in xml_mapped: cv2.circle(xml_gt_img,(int(px),int(py)),5,(0,0,255),-1)
|
| 89 |
|
| 90 |
+
# Convert to RGB (keep original size)
|
|
|
|
| 91 |
flat_rgb = cv2.cvtColor(flat_img,cv2.COLOR_BGR2RGB)
|
|
|
|
| 92 |
match_rgb = cv2.cvtColor(match_img,cv2.COLOR_BGR2RGB)
|
|
|
|
| 93 |
roi_rgb = cv2.cvtColor(persp_roi,cv2.COLOR_BGR2RGB)
|
|
|
|
| 94 |
xml_rgb = cv2.cvtColor(xml_gt_img,cv2.COLOR_BGR2RGB)
|
|
|
|
| 95 |
|
| 96 |
+
# Merge 2x2 grid (maintaining original resolution)
|
| 97 |
top = np.hstack([flat_rgb, match_rgb])
|
| 98 |
bottom = np.hstack([roi_rgb, xml_rgb])
|
| 99 |
combined_grid = np.vstack([top,bottom])
|