Skip to content

Commit

Permalink
adding config files for larc2023
Browse files Browse the repository at this point in the history
  • Loading branch information
igormaurell committed Oct 8, 2023
1 parent 30bd865 commit 29a6db5
Show file tree
Hide file tree
Showing 3 changed files with 27 additions and 16 deletions.
2 changes: 1 addition & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
*.lua
*.csv
*.dat
*.pt
*.caffemodel
*.pt
*.pbtxt
*.prototxt
39 changes: 25 additions & 14 deletions butia_recognition/config/yolov8_object_recognition_custom.yaml
Original file line number Diff line number Diff line change
@@ -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:

Expand Down
2 changes: 1 addition & 1 deletion butia_recognition/launch/yolov8_object_recognition.launch
Original file line number Diff line number Diff line change
Expand Up @@ -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)"/>

Expand Down

0 comments on commit 29a6db5

Please sign in to comment.