Browse Source

Update 3d_accuracy.py

Kenkart 1 year ago
parent
commit
64013a9976
1 changed files with 3 additions and 2 deletions
  1. 3 2
      3d_accuracy.py

+ 3 - 2
3d_accuracy.py

@@ -23,7 +23,8 @@ def vector_string_to_float(vector):
 
 fig, ax = plt.subplots(1, 3, sharey=True)
 
-for root, dir, files in os.walk(os.path.join(os.getcwd(), 'DataCSV\\Pieter\\')):
+name = 'Pieter'
+for root, dir, files in os.walk(os.path.join(os.getcwd(), f'DataCSV\\{name}\\')):
     for file in files:
         if file.endswith(".csv"):
             path = os.path.join(root, file)
@@ -41,7 +42,7 @@ for root, dir, files in os.walk(os.path.join(os.getcwd(), 'DataCSV\\Pieter\\')):
             y = [sum_distance_joints[i] / counter for i in range(18)]
             if "FirstPerson" in file:
                 ax[0].scatter(x, y)
-            if "ThirdPersonPerspective_" in file:
+            elif "ThirdPersonPerspective_" in file:
                 ax[1].scatter(x, y)
             elif "MultipleViews" in file:
                 ax[2].scatter(x, y)