Bart/bullet3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update kuka_diverse_object_gym_env.py

Browse Source 
Fix issue in glob to separate training/test data pattern. See https://github.com/bulletphysics/bullet3/issues/2158
Thank to @TomoyaFukui
This commit is contained in:
erwincoumans
2019-03-14 09:33:48 -07:00
committed by GitHub
parent 18918389d3
commit 9662ae4ba0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py
View File

@@ -316,7 +316,7 @@ class KukaDiverseObjectEnv(KukaGymEnv):
if test: if test:
urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*0/*.urdf') urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*0/*.urdf')
else: else:
urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*[^0]/*.urdf') urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*[1-9]/*.urdf')
found_object_directories = glob.glob(urdf_pattern) found_object_directories = glob.glob(urdf_pattern)
total_num_objects = len(found_object_directories) total_num_objects = len(found_object_directories)
selected_objects = np.random.choice(np.arange(total_num_objects), selected_objects = np.random.choice(np.arange(total_num_objects),