Browse Source

fix 3d_accuracy.py

Kenkart 2 years ago
parent
commit
4c817c1b75
1 changed files with 15 additions and 7 deletions
  1. 15 7
      3d_accuracy.py

+ 15 - 7
3d_accuracy.py

@@ -40,13 +40,16 @@ for root, dir, files in os.walk(os.path.join(os.getcwd(), 'DataCSV\\Gary\\')):
             x = [header[i][5:] for i in range(18)]
             y = [sum_distance_joints[i] / counter for i in range(18)]
             if "FirstPerson" in file:
-                sc = ax[0].scatter(x, y)
-                # for i in range(len(x)):
-                #     ax[0].annotate(file[22:], xy=(i,y[i]))
+                ax[0].scatter(x, y)
             if "ThirdPersonPerspective_" in file:
                 ax[1].scatter(x, y)
             elif "MultipleViews" in file:
                 ax[2].scatter(x, y)
+            # Reset values
+            sum_distance_joints = np.zeros(18)
+            counter = 0
+
+
 fig.autofmt_xdate(rotation=45)
 ax[0].set_title('First Person')
 ax[1].set_title('Third Person')
@@ -56,8 +59,14 @@ ax[2].set_title('Third Person Multiple View')
 # plt.legend(files).set_draggable(True)
 plt.show()
 
-path = os.path.join(os.getcwd(
-), 'DataCSV\\name\\FirstPersonPerspective_OneArm_Forward_HapticFeedback_Slow.csv')
+
+# ----------------------------------------
+
+
+sum_distance_joints = np.zeros(18)
+counter = 0
+
+path = os.path.join(os.getcwd(), 'DataCSV\\Gary\\FirstPersonPerspective_OneArm_Backward_ColorFeedback_Fast.csv')
 with open(path, newline='') as csvfile:
     reader = csv.reader(csvfile)
     header = next(reader)
@@ -70,5 +79,4 @@ with open(path, newline='') as csvfile:
         counter += 1
 
 for i in range(sum_distance_joints.size):
-    print("3d accuracy ", header[i][5:], ": ",
-          sum_distance_joints[i] / counter)
+    print("3d accuracy ", header[i][5:], ": ", sum_distance_joints[i] / counter)