in data_loaders.py [0:0]
def get_training_instance(self):
# select an object and and a principle grasp randomly
class_choice = random.choice(self.classes_names)
object_choice = random.choice(class_choice)
obj, obj_class = object_choice
# select the remaining grasps and shuffle the select grasps
num_choices = [0, 1, 2, 3, 4]
nums = []
for i in range(self.args.num_grasps):
choice = random.choice(num_choices)
nums.append(choice)
del (num_choices[num_choices.index(choice)])
random.shuffle(nums)
return obj, obj_class, nums[-1], nums