|
255 | 255 | "entity_cfg": {"uid": "cam_right_wrist"}, |
256 | 256 | "foreground_uids": ["bottle", "cup"] |
257 | 257 | } |
258 | | - }, |
259 | | - "exteroception": { |
260 | | - "func": "compute_exteroception", |
261 | | - "mode": "add", |
262 | | - "name": "exteroception", |
263 | | - "params": { |
264 | | - "descriptor": { |
265 | | - "all_sensors": [ |
266 | | - { |
267 | | - "type": "robot", |
268 | | - "control_part": "left_arm", |
269 | | - "offset": [0.0, 0.0, 0.025], |
270 | | - "follow_eef": true |
271 | | - }, |
272 | | - { |
273 | | - "type": "robot", |
274 | | - "control_part": "right_arm", |
275 | | - "offset": [0.0, 0.0, 0.025], |
276 | | - "follow_eef": true |
277 | | - }, |
278 | | - { |
279 | | - "type": "affordance", |
280 | | - "obj_uid": "bottle", |
281 | | - "key": "bottle_grasp_pose_object_unoffset", |
282 | | - "is_arena_coord": false, |
283 | | - "follow_eef": "right_eef" |
284 | | - }, |
285 | | - { |
286 | | - "type": "affordance", |
287 | | - "obj_uid": "bottle", |
288 | | - "key": "pour_water_start_pose_unoffset", |
289 | | - "is_arena_coord": true, |
290 | | - "follow_eef": "right_eef" |
291 | | - }, |
292 | | - { |
293 | | - "type": "affordance", |
294 | | - "obj_uid": "bottle", |
295 | | - "key": "pour_water_stop_pose_unoffset", |
296 | | - "is_arena_coord": true, |
297 | | - "follow_eef": "right_eef" |
298 | | - }, |
299 | | - { |
300 | | - "type": "affordance", |
301 | | - "obj_uid": "bottle", |
302 | | - "key": "bottle_place_pose_unoffset", |
303 | | - "is_arena_coord": true |
304 | | - } |
305 | | - ] |
306 | | - }, |
307 | | - "x_interval": 0.02, |
308 | | - "y_interval": 0.05, |
309 | | - "kpnts_number": 2, |
310 | | - "groups": 6 |
311 | | - } |
312 | 258 | } |
313 | 259 | }, |
314 | 260 | "dataset": { |
315 | | - "instruction": { |
316 | | - "lang": "Pour water from the bottle into the mug." |
317 | | - }, |
318 | 261 | "robot_meta": { |
319 | 262 | "arm_dofs": 12, |
320 | 263 | "control_freq": 25, |
321 | 264 | "control_parts": ["left_arm", "left_eef", "right_arm", "right_eef"], |
322 | | - "observation": { |
323 | | - "vision": { |
324 | | - "cam_high": ["mask", "exteroception"], |
325 | | - "cam_right_wrist": ["mask", "exteroception"], |
326 | | - "cam_left_wrist": ["mask", "exteroception"] |
327 | | - }, |
328 | | - "states": ["qpos"], |
329 | | - "exteroception": ["cam_high", "cam_right_wrist", "cam_left_wrist"] |
330 | | - }, |
331 | 265 | "min_len_steps": 5 |
332 | 266 | } |
333 | | - }, |
334 | | - "success_params": { |
335 | | - "strict": false |
336 | 267 | } |
337 | 268 | }, |
338 | 269 | "robot": { |
|
0 commit comments