From 7288c466173018a1461c65f440ef994d264b373a Mon Sep 17 00:00:00 2001 From: Miguel Date: Mon, 6 May 2024 09:14:44 +0200 Subject: [PATCH] Update weed and crop category names in System class --- field_friend/system.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/field_friend/system.py b/field_friend/system.py index 87a02401..924c8aa0 100644 --- a/field_friend/system.py +++ b/field_friend/system.py @@ -72,9 +72,9 @@ def watch_robot() -> None: self.kpi_provider.increment_on_rising_edge('low_battery', self.field_friend.bms.is_below_percent(10.0)) self.puncher = Puncher(self.field_friend, self.driver, self.kpi_provider) - self.big_weed_category_names = ['thistle', 'big_weed', 'orache'] - self.small_weed_category_names = ['weed', 'coin'] - self.crop_category_names = ['sugar_beet', 'crop', 'coin_with_hole'] + self.big_weed_category_names = ['big_weed', 'thistle', 'orache', 'weedy_area', ] + self.small_weed_category_names = ['coin', 'weed',] + self.crop_category_names = ['coin_with_hole', 'crop', 'sugar_beet', 'onion', 'garlic', ] self.plant_locator = PlantLocator(self.usb_camera_provider, self.detector, self.plant_provider, @@ -82,7 +82,7 @@ def watch_robot() -> None: ) self.plant_locator.weed_category_names = self.big_weed_category_names + self.small_weed_category_names self.plant_locator.crop_category_names = self.crop_category_names - self.plant_locator.minimum_weed_confidence = 0.8 + self.plant_locator.minimum_weed_confidence = 0.85 self.plant_locator.minimum_crop_confidence = 0.40 rosys.on_repeat(watch_robot, 1.0)