def construct_training_scene():
speaker = Speaker(Vec2(0, 0))
scene = Scene(3)
table = Landmark(
'table',
RectangleRepresentation(
rect=BoundingBox([Vec2(-0.4, 0.4), Vec2(0.4, 1.0)])), None,
ObjectClass.TABLE)
obj1 = Landmark(
'green_cup',
RectangleRepresentation(rect=BoundingBox(
[Vec2(0.05 - 0.035, 0.9 - 0.035),
Vec2(0.05 + 0.035, 0.9 + 0.035)]),
landmarks_to_get=[]), None, ObjectClass.CUP,
Color.GREEN)
obj2 = Landmark(
'blue_cup',
RectangleRepresentation(rect=BoundingBox(
[Vec2(0.05 - 0.035, 0.7 - 0.035),
Vec2(0.05 + 0.035, 0.7 + 0.035)]),
landmarks_to_get=[]), None, ObjectClass.CUP,
Color.BLUE)
obj3 = Landmark(
'pink_cup',
RectangleRepresentation(rect=BoundingBox(
[Vec2(0 - 0.035, 0.55 - 0.035),
Vec2(0 + 0.035, 0.55 + 0.035)]),
landmarks_to_get=[]), None, ObjectClass.CUP,
Color.PINK)
obj4 = Landmark(
'purple_prism',
RectangleRepresentation(rect=BoundingBox(
[Vec2(-0.3 - 0.03, 0.7 - 0.05),
Vec2(-0.3 + 0.03, 0.7 + 0.05)]),
landmarks_to_get=[]), None, ObjectClass.PRISM,
Color.PURPLE)
obj5 = Landmark(
'orange_prism',
RectangleRepresentation(rect=BoundingBox(
[Vec2(0.3 - 0.03, 0.7 - 0.05),
Vec2(0.3 + 0.03, 0.7 + 0.05)]),
landmarks_to_get=[]), None, ObjectClass.PRISM,
Color.ORANGE)
scene.add_landmark(table)
for obj in (obj1, obj2, obj3, obj4, obj5):
obj.representation.alt_representations = []
scene.add_landmark(obj)
return scene, speaker
def construct_training_scene():
speaker = Speaker(Vec2(0,0))
scene = Scene(3)
table = Landmark('table',
RectangleRepresentation(rect=BoundingBox([Vec2(-0.4,0.4), Vec2(0.4,1.0)])),
None,
ObjectClass.TABLE)
obj1 = Landmark('green_cup',
RectangleRepresentation(rect=BoundingBox([Vec2(0.05-0.035,0.9-0.035), Vec2(0.05+0.035,0.9+0.035)]), landmarks_to_get=[]),
None,
ObjectClass.CUP,
Color.GREEN)
obj2 = Landmark('blue_cup',
RectangleRepresentation(rect=BoundingBox([Vec2(0.05-0.035,0.7-0.035), Vec2(0.05+0.035,0.7+0.035)]), landmarks_to_get=[]),
None,
ObjectClass.CUP,
Color.BLUE)
obj3 = Landmark('pink_cup',
RectangleRepresentation(rect=BoundingBox([Vec2(0-0.035,0.55-0.035), Vec2(0+0.035,0.55+0.035)]), landmarks_to_get=[]),
None,
ObjectClass.CUP,
Color.PINK)
obj4 = Landmark('purple_prism',
RectangleRepresentation(rect=BoundingBox([Vec2(-0.3-0.03,0.7-0.05), Vec2(-0.3+0.03,0.7+0.05)]), landmarks_to_get=[]),
None,
ObjectClass.PRISM,
Color.PURPLE)
obj5 = Landmark('orange_prism',
RectangleRepresentation(rect=BoundingBox([Vec2(0.3-0.03,0.7-0.05), Vec2(0.3+0.03,0.7+0.05)]), landmarks_to_get=[]),
None,
ObjectClass.PRISM,
Color.ORANGE)
obj6 = Landmark('narrow_prism',
RectangleRepresentation(rect=BoundingBox([Vec2(0.2-0.04,0.5-0.02), Vec2(0.4+0.03,0.4+0.03)]), landmarks_to_get=[]),
None,
ObjectClass.PRISM,
Color.ORANGE)
scene.add_landmark(table)
for obj in (obj1, obj2, obj3, obj4, obj5,obj6):
obj.representation.alt_representations = []
scene.add_landmark(obj)
return scene, speaker
obj1 = Landmark('obj1',
RectangleRepresentation(rect=BoundingBox([Vec2(5,5), Vec2(5.1,5.1)])),
None,
Landmark.CUP)
obj2 = Landmark('obj2',
RectangleRepresentation(rect=BoundingBox([Vec2(5.5,6), Vec2(5.6,6.1)])),
None,
Landmark.BOTTLE)
obj3 = Landmark('obj3',
RectangleRepresentation(rect=BoundingBox([Vec2(4.5,4.5), Vec2(4.8,4.8)])),
None,
Landmark.CHAIR)
scene.add_landmark(table)
scene.add_landmark(obj1)
scene.add_landmark(obj2)
# scene.add_landmark(obj3)
# groups = adapter.adapt(scene)
# for i,g in enumerate(groups):
# if (len(g) > 1):
# scene.add_landmark(Landmark('ol%d'%i, ObjectLineRepresentation(g), None, Landmark.LINE))
# f = open('scene.pickle','wb')
# pickle.dump(scene,f)
# f.flush()
# f.close()
# del scene
