Fix(bop_challenge_tless): sample without replacement

ompugao · Sep 19, 2022 · 6f47392 · 6f47392
1 parent adee42b
commit 6f47392
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/datasets/bop_challenge/main_tless_random.py b/examples/datasets/bop_challenge/main_tless_random.py
@@ -63,7 +63,7 @@ def sample_pose_func(obj: bproc.types.MeshObject):
 for i in range(args.num_scenes):
 
     # Sample bop objects for a scene
-    sampled_target_bop_objs = list(np.random.choice(target_bop_objs, size=20))
+    sampled_target_bop_objs = list(np.random.choice(target_bop_objs, size=20, replace=False))
     sampled_distractor_bop_objs = list(np.random.choice(itodd_dist_bop_objs, size=2, replace=False))
     sampled_distractor_bop_objs += list(np.random.choice(ycbv_dist_bop_objs, size=2, replace=False))
     sampled_distractor_bop_objs += list(np.random.choice(hb_dist_bop_objs, size=2, replace=False))