EmbodiedAI
/
RoboWaiter
mirror of https://github.com/HPCL-EI/RoboWaiter.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
RoboWaiter/robowaiter/llm_client/find_obj_utils.py

121 lines
8.7 KiB
Python
Raw Normal View History

tool_api
2023-11-17 16:22:31 +08:00
import random
# all_obj_en = ['Mug', 'Banana', 'Toothpaste', 'Bread', 'Softdrink', 'Yogurt', 'ADMilk', 'VacuumCup', 'BottledDrink', 'PencilVase',
# 'Teacup', 'Caddy', 'Dictionary', 'Cake', 'Date', 'Stapler', 'LunchBox', 'Bracelet', 'MilkDrink', 'CocountWater', 'Walnut', 'HamSausage',
# 'GlueStick', 'AdhensiveTape', 'Calculator', 'Chess', 'Orange', 'Glass', 'Washbowl', 'Durian', 'Gum', 'Towl', 'OrangeJuice', 'Cardcase',
# 'RubikCube', 'StickyNotes', 'NFCJuice', 'SpringWater', 'Apple', 'Coffee', 'Gauze', 'Mangosteen', 'SesameSeedCake', 'Glove', 'Mouse',
# 'Kettle', 'Atomize', 'Chips', 'SpongeGourd', 'Garlic', 'Potato', 'Tray', 'Hemomanometer', 'TennisBall', 'ToyDog', 'ToyBear', 'TeaTray',
# 'Sock', 'Scarf', 'ToiletPaper', 'Milk', 'Soap', 'Novel', 'Watermelon', 'Tomato', 'CleansingFoam', 'CocountMilk', 'SugarlessGum',
# 'MedicalAdhensiveTape', 'SourMilkDrink', 'PaperCup', 'Tissue', 'YogurtDrink', 'Newspaper', 'Box', 'PaperCupStarbucks', 'CoffeeMachine',
# 'Straw', 'Cake', 'Tray', 'Bread', 'Glass', 'Door', 'Mug', 'Machine', 'PackagedCoffee', 'CubeSugar',
# 'Apple', 'Spoon', 'Drinks', 'Drink', 'Take-AwayCup', 'Saucer', 'TrashBin', 'Knife', 'Ginger', 'Floor', 'Roof', 'Wall'] + \
# ['Broom', 'CoffeeCup', 'Door', 'Glass', 'Mug', 'ZhuZi', 'Towel', 'LaJiTong', 'CoffeeMachine', 'ZaoTai', 'Sofa', 'Cake', 'ChaTou',
# 'Plate', 'CoffeeBag', 'TuoBu', 'KaiGuan', 'Kettle', 'Apple', 'ChaZuo', 'Sugar', 'BaTai', 'BaoJing', 'DrinkMachine', 'KongTiao',
# 'Desk', 'Clip', 'Knife', 'TuoPan', 'BoJi', 'ZhiBeiHe', 'Bread', 'WaterCup', 'Box', 'Chair', 'Hand', 'XiGuan', 'Spoon', 'Container',
# 'IceMachine', 'KaoXiang', 'SaoBa', 'XiangGui']
# , '机器人左手', '机器人右手', 'Bernachon'
all_obj = ['马克杯', '香蕉', '牙膏', '面包', '软饮料', '酸奶', 'AD钙奶', '真空杯', '瓶装饮料', '铅笔花瓶', '茶杯', '茶匙', '词典',
'蛋糕', '日期', '订书机', '午餐盒', '手镯', '牛奶饮品', '椰水', '核桃', '火腿肠', '胶棒', '胶带', '计算器', '国际象棋', '橙子',
'玻璃杯', '洗碗盆', '榴莲', '口香糖', '毛巾', '橙汁', '卡片盒', '魔方', '便签', 'NFC汁', '矿泉水', '苹果', '咖啡', '纱布', '芒果',
'芝麻蛋糕', '手套', '鼠标', '水壶', '雾化器', '薯片', '丝瓜', '大蒜', '土豆', '托盘', '血压计', '网球', '玩具狗', '玩具熊', '茶盘',
'袜子', '围巾', '卫生纸', '牛奶', '肥皂', '小说', '西瓜', '番茄', '洁面泡沫', '椰奶', '无糖口香糖', '医用胶带', '酸奶饮品', '纸杯',
'纸巾', '酸奶饮品', '报纸', '盒子', '星巴克纸杯', '咖啡机', '吸管', '蛋糕', '托盘', '面包', '玻璃杯', '门',
'马克杯', '机器', '包装咖啡', '方糖', '苹果', '勺子', '饮料', '饮料', '外带杯', '茶杯', '垃圾桶', '刀', '姜', '地板', '屋顶', '墙'] + \
['扫帚', '咖啡杯', '门', '玻璃杯', '马克杯', '著子', '毛巾', '垃圾桶', '咖啡机', '灶台', '沙发', '蛋糕', '插头', '盘子', '咖啡袋', '拖布', '开关',
'水壶', '苹果', '插座', '糖', '吧台', '报警', '饮料机', '空调', '桌子', '夹子', '刀', '托盘', '簸箕', '纸杯盒', '面包', '水杯', '盒子', '椅子',
'手', '吸管', '勺子', '容器', '制冰机', '烤箱', '扫把', '香柜']
# 储藏室
all_loc = ['吧台', '餐桌', '沙发', '灶台', '大门', '灯开关', '空调开关', '橱柜', '卫生间', '窗户', '音响', '休闲区', '工作台', '服务台', '收银台', '墙角',
'蛋糕柜', '充电处', '冰箱', '书架']
更新工具调用
2023-11-19 14:21:58 +08:00
all_loc_en = ['bar', 'Table', 'sofa', 'stove', 'Gate', 'light switch', 'airconditioner switch', 'cabinet', 'bathroom', 'window','audio', 'lounge area',
'workstation', 'service counter', 'cashier counter', 'corner','cake display', 'ChargingStations', 'refrigerator', 'bookshelf']
tool_api
2023-11-17 16:22:31 +08:00
loc_map_en = {'bar': {'工作台', '服务台', '收银台', '蛋糕柜'},
Update
2023-11-17 21:17:19 +08:00
'Table': {'大门', '休闲区', '墙角'},
tool_api
2023-11-17 16:22:31 +08:00
'sofa': {'餐桌', '窗户', '音响', '休闲区', '墙角', '书架'},
'stove': {'吧台', '橱柜', '工作台', '服务台', '收银台', '蛋糕柜', '冰箱'},
'Gate': {'吧台', '灯开关', '空调开关', '卫生间', '墙角'},
'light switch': {'大门', '空调开关', '卫生间', '墙角'},
'airconditioner switch': {'大门', '灯开关', '卫生间', '墙角'},
'cabinet': {'灶台', '吧台', '工作台', '服务台', '收银台', '蛋糕柜', '充电处', '冰箱'},
'bathroom': {'大门', '墙角'},
'window': {'餐桌', '沙发', '休闲区'},
'audio': {'餐桌', '沙发', '休闲区', '墙角', '书架'},
'lounge area': {'沙发', '餐桌', '墙角', '书架', '音响'},
'workstation': {'吧台', '服务台', '收银台'},
'service counter': {'吧台', '工作台', '收银台'},
'cashier counter': {'吧台', '工作台', '服务台'},
'corner': {'卫生间', '沙发', '灯开关', '空调开关', '音响', '休闲区', '书架'},
'cake display': {'吧台', '橱柜', '服务台', '收银台', '冰箱'},
更新了地点的工具调用
2023-11-17 17:13:27 +08:00
'ChargingStations': {'吧台', '餐桌', '沙发', '休闲区', '工作台', '服务台', '收银台', '墙角', '书架'},
tool_api
2023-11-17 16:22:31 +08:00
'refrigerator': {'吧台', '服务台', '蛋糕柜'},
'bookshelf': {'餐桌', '沙发', '窗户', '休闲区', '墙角'}
}
loc_map = {'吧台': {'灶台', '橱柜', '工作台', '服务台', '收银台', '蛋糕柜', '充电处', '冰箱'},
'餐桌': {'沙发', '大门', '窗户', '休闲区', '墙角', '椅子', '书架'},
'沙发': {'餐桌', '窗户', '音响', '休闲区', '墙角', '书架'},
'灶台': {'吧台', '橱柜', '工作台', '服务台', '收银台', '蛋糕柜', '冰箱'},
'大门': {'吧台', '灯开关', '空调开关', '卫生间', '墙角'},
'灯开关': {'大门', '空调开关', '卫生间', '墙角'},
'空调开关': {'大门', '灯开关', '卫生间', '墙角'},
'橱柜': {'灶台', '吧台', '工作台', '服务台', '收银台', '蛋糕柜', '充电处', '冰箱'},
'卫生间': {'大门', '墙角'},
'窗户': {'餐桌', '沙发', '休闲区'},
'音响': {'餐桌', '沙发', '休闲区', '墙角', '书架'},
'休闲区': {'沙发', '餐桌', '墙角', '书架', '音响'},
'工作台': {'吧台', '服务台', '收银台'},
'服务台': {'吧台', '工作台', '收银台'},
'收银台': {'吧台', '工作台', '服务台'},
'墙角': {'卫生间', '沙发', '灯开关', '空调开关', '音响', '休闲区', '书架'},
'蛋糕柜': {'吧台', '橱柜', '服务台', '收银台', '冰箱'},
'充电处': {'吧台', '餐桌', '沙发', '休闲区', '工作台', '服务台', '收银台', '墙角', '书架'},
'冰箱': {'吧台', '灶台', '工作台', '服务台', '收银台', '蛋糕柜'},
'书架': {'餐桌', '沙发', '窗户', '休闲区', '墙角'}
}
# print("looc_map:", loc_map['吧台'])
# print("looc_map:", random.choice(list(loc_map['吧台'])))
# import synonyms
# def get_synonyms(word):
# synonyms_list = synonyms.nearby(word)[0]
# return synonyms_list
#
# target_word = "高兴"
# synonyms_result = get_synonyms(target_word)
# print("synonyms_result:",synonyms_result)
# def save_variable(self):
# scene = self.status
# init_obj_name_set = set()
# init_obj_info_dict = {}
# for item in scene.objects:
# name = item.name
# location = item.location
# init_obj_name_set.add(name)
# # 如果name已经存在，则添加一个唯一标识符
# if name in init_obj_info_dict:
# count = 1
# new_name = f"{name}_{count}"
# while new_name in init_obj_info_dict:
# count += 1
# new_name = f"{name}_{count}"
# name = new_name
# init_obj_info_dict[name] = location
# # print("obj_info_dict:",obj_info_dict)
# # print("obj_name_set:",obj_name_set)
# import os
# import pickle
# absolute_path = r"E:\DATA\Projects\Robot1002 5.2\Plugins\LLM_prompt\data\init_obj_name_set.pkl"
# target_directory = os.path.dirname(absolute_path)
# os.makedirs(target_directory, exist_ok=True)
# # 将变量保存到二进制文件
# with open(absolute_path, 'wb') as file:
# pickle.dump(init_obj_name_set, file)