axis label

app.py

@@ -169,9 +169,23 @@ def infer_func(img, do_rm_bkg):
     # 旋转后的向量
     rot_x, rot_y, rot_z = get_proj2D_XYZ(phi, theta, gamma)
 
-    draw_axis(ax, origin, rot_y, 'g')
-    draw_axis(ax, origin, rot_z, 'b')
-    draw_axis(ax, origin, rot_x, 'r')
+    # draw arrow
+    arrow_attr = [{'point':rot_x, 'color':'r', 'label':'front'}, 
+                  {'point':rot_y, 'color':'g', 'label':'right'}, 
+                  {'point':rot_z, 'color':'b', 'label':'top'}]
+    
+    if phi> 45 and phi<=225:
+        order = [0,1,2]
+    elif phi > 225 and phi < 315:
+        order = [2,0,1]
+    else:
+        order = [2,1,0]
+    
+    for i in range(3):
+        draw_axis(ax, origin, arrow_attr[order[i]]['point'], arrow_attr[order[i]]['color'], arrow_attr[order[i]]['label'])
+        # draw_axis(ax, origin, rot_y, 'g', label='right')
+        # draw_axis(ax, origin, rot_z, 'b', label='top')
+        # draw_axis(ax, origin, rot_x, 'r', label='front')
 
     # 关闭坐标轴和网格
     ax.set_axis_off()