diff --git a/README.md b/README.md index ba54228..a70c964 100644 --- a/README.md +++ b/README.md @@ -44,7 +44,7 @@ ## 功能需求 点位1:识别 井盖(塌陷 移位 破损 缺失,都认为是完整),井眼,发送位置让机械臂转动, 报【井盖完整,燃气浓度正常/异常】 -点位2:识别 引入口(目前没识别),发送位置让机械臂转动,报 【XXXX,燃气浓度正常/异常】 +点位2:识别 引入口,发送位置让机械臂转动,报 【XXXX,燃气浓度正常/异常】 点位3:识别 调压箱(目前没识别),???,报【XXXX,燃气浓度正常/异常】 以上都是机器狗走到目标点位,才执行 diff --git a/README.md b/README.md index ba54228..a70c964 100644 --- a/README.md +++ b/README.md @@ -44,7 +44,7 @@ ## 功能需求 点位1:识别 井盖(塌陷 移位 破损 缺失,都认为是完整),井眼,发送位置让机械臂转动, 报【井盖完整,燃气浓度正常/异常】 -点位2:识别 引入口(目前没识别),发送位置让机械臂转动,报 【XXXX,燃气浓度正常/异常】 +点位2:识别 引入口,发送位置让机械臂转动,报 【XXXX,燃气浓度正常/异常】 点位3:识别 调压箱(目前没识别),???,报【XXXX,燃气浓度正常/异常】 以上都是机器狗走到目标点位,才执行 diff --git a/config.py b/config.py index 658edb5..0eed53b 100644 --- a/config.py +++ b/config.py @@ -39,8 +39,8 @@ { "cam_id": 0, "gst_str": 0, - "tcp_send_cls": ["井盖眼"], - "alarm_send_cls": ["路锥", "反光衣"], + "tcp_send_cls": ["井盖眼"], # todo 要加引入口吗? + "alarm_send_cls": ["路锥", "反光衣", "围挡"], "frame_interval": 5, "remark": "本地测试摄像头", "save_annotated_images": True, @@ -70,18 +70,20 @@ 6: '人', 7: '压路机', 8: '反光衣', - 9: '土堆', - 10: '土方车', - 11: '头', - 12: '安全帽', - 13: '挖掘机', - 14: '推土机', - 15: '施工路牌', - 16: '水马', - 17: '路锥', - 18: '铁锹', - 19: '防护栏', - 20: '风镐' + 9: '围挡', + 10: '土堆', + 11: '土方车', + 12: '头', + 13: '安全帽', + 14: '引入口', + 15: '挖掘机', + 16: '推土机', + 17: '施工路牌', + 18: '水马', + 19: '路锥', + 20: '铁锹', + 21: '防护栏', + 22: '风镐' } MODEL = { @@ -93,9 +95,9 @@ BIZ_CLASS = { "WELL": ['井盖', '井盖塌陷', '井盖眼', '井盖破损', '井盖移位', '井盖缺失', ], - "ENTER": [], + "ENTER": ['引入口'], "BOX": [], - "CONSTRUCTION": ['反光衣', '路锥'] + "CONSTRUCTION": ['反光衣', '路锥', '围挡'] } diff --git a/README.md b/README.md index ba54228..a70c964 100644 --- a/README.md +++ b/README.md @@ -44,7 +44,7 @@ ## 功能需求 点位1:识别 井盖(塌陷 移位 破损 缺失,都认为是完整),井眼,发送位置让机械臂转动, 报【井盖完整,燃气浓度正常/异常】 -点位2:识别 引入口(目前没识别),发送位置让机械臂转动,报 【XXXX,燃气浓度正常/异常】 +点位2:识别 引入口,发送位置让机械臂转动,报 【XXXX,燃气浓度正常/异常】 点位3:识别 调压箱(目前没识别),???,报【XXXX,燃气浓度正常/异常】 以上都是机器狗走到目标点位,才执行 diff --git a/config.py b/config.py index 658edb5..0eed53b 100644 --- a/config.py +++ b/config.py @@ -39,8 +39,8 @@ { "cam_id": 0, "gst_str": 0, - "tcp_send_cls": ["井盖眼"], - "alarm_send_cls": ["路锥", "反光衣"], + "tcp_send_cls": ["井盖眼"], # todo 要加引入口吗? + "alarm_send_cls": ["路锥", "反光衣", "围挡"], "frame_interval": 5, "remark": "本地测试摄像头", "save_annotated_images": True, @@ -70,18 +70,20 @@ 6: '人', 7: '压路机', 8: '反光衣', - 9: '土堆', - 10: '土方车', - 11: '头', - 12: '安全帽', - 13: '挖掘机', - 14: '推土机', - 15: '施工路牌', - 16: '水马', - 17: '路锥', - 18: '铁锹', - 19: '防护栏', - 20: '风镐' + 9: '围挡', + 10: '土堆', + 11: '土方车', + 12: '头', + 13: '安全帽', + 14: '引入口', + 15: '挖掘机', + 16: '推土机', + 17: '施工路牌', + 18: '水马', + 19: '路锥', + 20: '铁锹', + 21: '防护栏', + 22: '风镐' } MODEL = { @@ -93,9 +95,9 @@ BIZ_CLASS = { "WELL": ['井盖', '井盖塌陷', '井盖眼', '井盖破损', '井盖移位', '井盖缺失', ], - "ENTER": [], + "ENTER": ['引入口'], "BOX": [], - "CONSTRUCTION": ['反光衣', '路锥'] + "CONSTRUCTION": ['反光衣', '路锥', '围挡'] } diff --git a/weights/go-v8m-20250611.pt b/weights/go-v8m-20250611.pt new file mode 100644 index 0000000..48c5b38 --- /dev/null +++ b/weights/go-v8m-20250611.pt Binary files differ