martin/object-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small update

Browse Source
This commit is contained in:
Martin Laasmaa
2026-01-16 10:39:46 +02:00
parent eefda5b878
commit 506c74e53a
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/show_yolo_seg.py
View File

@@ -196,7 +196,9 @@ def main():
bbox = np.array(bbox) * np.array([img.shape[1], img.shape[0], img.shape[1], img.shape[0]]) bbox = np.array(bbox) * np.array([img.shape[1], img.shape[0], img.shape[1], img.shape[0]])
yc, xc, h, w = bbox yc, xc, h, w = bbox
print("bbox", bbox) print("bbox", bbox)
polyline = np.array(coords[4:]).reshape(-1, 2) * np.array([img.shape[1], img.shape[0]])
# polyline = np.array(coords[4:]).reshape(-1, 2) * np.array([img.shape[1], img.shape[0]])
polyline = np.array(coords).reshape(-1, 2) * np.array([img.shape[1], img.shape[0]])
print("pl", coords[4:]) print("pl", coords[4:])
print("pl", polyline) print("pl", polyline)
@@ -207,12 +209,13 @@ def main():
plt.figure(figsize=(10, 10 * out.shape[0] / out.shape[1])) plt.figure(figsize=(10, 10 * out.shape[0] / out.shape[1]))
plt.imshow(out_rgb) plt.imshow(out_rgb)
plt.plot(polyline[:, 0], polyline[:, 1], "y", linewidth=2) plt.plot(polyline[:, 0], polyline[:, 1], "y", linewidth=2)
plt.plot( if 0:
[yc - h / 2, yc - h / 2, yc + h / 2, yc + h / 2, yc - h / 2], plt.plot(
[xc - w / 2, xc + w / 2, xc + w / 2, xc - w / 2, xc - w / 2], [yc - h / 2, yc - h / 2, yc + h / 2, yc + h / 2, yc - h / 2],
"r", [xc - w / 2, xc + w / 2, xc + w / 2, xc - w / 2, xc - w / 2],
linewidth=2, "r",
) linewidth=2,
)
# plt.axis("off") # plt.axis("off")
plt.title(f"{img_path.name} ({lbl_path.name})") plt.title(f"{img_path.name} ({lbl_path.name})")