From 3099de4baf3448ac0fe4e5dfdd3a3d894eeee8d2 Mon Sep 17 00:00:00 2001
From: maali <mahmoud-ahmed.ali@liris.cnrs.fr>
Date: Mon, 27 Mar 2023 16:51:01 +0200
Subject: [PATCH] call right function getAllValDataFruits
---
predict_pose.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/predict_pose.py b/predict_pose.py
index 46e3ad7..9686aad 100644
--- a/predict_pose.py
+++ b/predict_pose.py
@@ -153,7 +153,7 @@ if __name__ == '__main__':
basePath = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Evaluating/' + class_name
fps = np.loadtxt(f'Generated_Worlds_/Generated/{class_name}/{class_name}_fps_3d.txt')
- images_ls, labels_ls, mask_ls, choice_ls = data.getAllValData(class_name)
+ images_ls, labels_ls, mask_ls, choice_ls = data.getAllValDataFruits(class_name)
print(len(images_ls))
for i, img in enumerate(images_ls):
--
GitLab