diff --git a/predict_pose.py b/predict_pose.py index 46e3ad7d0b0d0e6246e4a594bb7b3d6818947866..9686aadcd6b51174c2de2886330d978ba2559b12 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):