From 29a6db5296a31bc7cf5a1dd9344ef395db6a7a91 Mon Sep 17 00:00:00 2001
From: Igor Maurell <igormaurell@gmail.com>
Date: Sun, 8 Oct 2023 01:44:22 -0300
Subject: [PATCH] adding config files for larc2023

---
 .gitignore                                    |  2 +-
 .../yolov8_object_recognition_custom.yaml     | 39 ++++++++++++-------
 .../launch/yolov8_object_recognition.launch   |  2 +-
 3 files changed, 27 insertions(+), 16 deletions(-)

diff --git a/.gitignore b/.gitignore
index fbeb6b74..4f2135d5 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,7 +6,7 @@
 *.lua
 *.csv
 *.dat
-*.pt
 *.caffemodel
+*.pt
 *.pbtxt
 *.prototxt
diff --git a/butia_recognition/config/yolov8_object_recognition_custom.yaml b/butia_recognition/config/yolov8_object_recognition_custom.yaml
index 0e23bffe..c6bfae36 100644
--- a/butia_recognition/config/yolov8_object_recognition_custom.yaml
+++ b/butia_recognition/config/yolov8_object_recognition_custom.yaml
@@ -1,24 +1,35 @@
 threshold: 0.3
 classes_by_category:
-    Food: ['pringles', 'flakes']
-    Drink: ['guarana', 'tonic']
-    Hygiene: ['cleaner']
+    Drinks: ['nescau', 'kuat', 'cononut water', 'fanta']
+    Cleaning supplies: ['detergent', 'sponge', 'cloth']
+    Pantry items: ['gelatin', 'mustard', 'shoyo', 'sauce', 'tea']
+    Fruits: ['apple', 'pear', 'tangerine']
+    Snacks: ['treloso', 'chocolate', 'peanut']
 
 all_classes:
-  - 'cleaner'
-  - 'flakes'
-  - 'guarana'
-  - 'pringles'
-  - 'tonic'
+  - 'apple'
+  - 'chocolate'
+  - 'cloth'
+  - 'cononut water'
+  - 'detergent'
+  - 'fanta'
+  - 'gelatin'
+  - 'kuat'
+  - 'mustard'
+  - 'nescau'
+  - 'peanut'
+  - 'pear'
+  - 'sauce'
+  - 'shoyo'
+  - 'sponge'
+  - 'tangerine'
+  - 'tea'
+  - 'treloso'
   
 max_sizes:
-    - [0.06, 0.255, 0.06]
-    - [0.08, 0.18, 0.14]
-    - [0.045, 0.15, 0.045]
-    - [0.07, 0.20, 0.07]
-    - [0.075, 0.125, 0.075]
+    - [0.05, 0.05, 0.05]
 
-model_file: yolov8-itec.pt
+model_file: yolov8-larc2023.pt
 
 subscribers:
   
diff --git a/butia_recognition/launch/yolov8_object_recognition.launch b/butia_recognition/launch/yolov8_object_recognition.launch
index 45ddd500..fdc0f7fe 100644
--- a/butia_recognition/launch/yolov8_object_recognition.launch
+++ b/butia_recognition/launch/yolov8_object_recognition.launch
@@ -3,7 +3,7 @@
   <arg name="use_machine" default="true"/>
   <arg name="output" default="screen"/>
   <arg name="node_name" default="butia_object_recognition"/>
-  <arg name="config_file" default="yolov8_object_recognition.yaml"/>
+  <arg name="config_file" default="yolov8_object_recognition_custom.yaml"/>
 
   <machine name="localhost" address="localhost" if="$(arg use_machine)"/>