Fixed X and Y axis in coco.py

mlcommons · Jan 5, 2025 · 5b9b622 · 5b9b622
1 parent a313db7
commit 5b9b622
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/plot.png b/plot.png
diff --git a/tools/upscale_coco/coco.py b/tools/upscale_coco/coco.py
@@ -317,7 +317,7 @@ def showAnns(self, anns):
                     v = kp[2::3]
                     for sk in sks:
                         if np.all(v[sk] > 0):
-                            plt.plot(x[sk], y[sk], linewidth=3, color=c)
+                            plt.plot(x[sk], y[sk], linewidth=3, color=c,label=f"keypoint {sk}")
                     plt.plot(
                         x[v > 0],
                         y[v > 0],
@@ -336,6 +336,10 @@ def showAnns(self, anns):
                         markeredgecolor=c,
                         markeredgewidth=2,
                     )
+                    plt.xlabel("X Coordinate")
+                    plt.ylabel("Y Coordinate") 
+                    print("Script is running correctly!")  
+                    plt.show()             
             p = PatchCollection(
                 polygons,
                 facecolor=color,