Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
40 commits
Select commit Hold shift + click to select a range
b5b5ec1
WIP: not compiling yet because of dependency
MatthijsBurgh Jun 24, 2025
e8a2678
(actions) add CI
MatthijsBurgh Jun 24, 2025
a1e06ed
Include with <> for onnx header
MatthijsBurgh Jun 24, 2025
42e6125
Updates gtests on the CMakeFile and made Preprocess public method
IasonTheodorou Jun 25, 2025
9d60544
Created initial tests and gitignore file
IasonTheodorou Jun 25, 2025
3914fc6
Created ONNX Yolo run tests
IasonTheodorou Jun 25, 2025
0280491
Renamed tests
IasonTheodorou Jun 25, 2025
2e9fb55
Small test update
IasonTheodorou Jun 25, 2025
38ad695
Use test Fixtures
IasonTheodorou Jun 25, 2025
723a133
Updated readme file
IasonTheodorou Jun 27, 2025
686ea7f
Commented out the classification parts
IasonTheodorou Jun 27, 2025
2cac602
Add inference engine calling functionality
IasonTheodorou Jul 1, 2025
a4038fc
Use inference to call the function from a single point
IasonTheodorou Jul 2, 2025
deb5227
Used Initilialize function to create a session outside the running se…
IasonTheodorou Jul 3, 2025
d5bcc12
Updated ReadmeFile
IasonTheodorou Jul 3, 2025
22da61d
Working switch between cuda and non cuda; code not working
MatthijsBurgh Sep 2, 2025
63e4f11
Made the yolo a catkin package
IasonTheodorou Sep 10, 2025
1742156
Renamed inference files to detection
IasonTheodorou Sep 10, 2025
ef1f775
Create single point of contact for Initialization and Detection (infe…
IasonTheodorou Sep 10, 2025
8a9cfc6
Refactored test according to detection.h way of inference
IasonTheodorou Sep 12, 2025
34a82df
Aligned Preprocessing and ONNX inference height width dimensions
IasonTheodorou Sep 12, 2025
2318110
Fixed the results vector in order not too keep boxes from previous im…
IasonTheodorou Sep 12, 2025
afb8972
Change CMake to be catkin package and small overall changes to run on…
IasonTheodorou Sep 12, 2025
06439ac
Merge remote-tracking branch 'origin/first_version' into feature/Refa…
MatthijsBurgh Sep 23, 2025
ec2c4c7
fix main.cpp
MatthijsBurgh Sep 30, 2025
266ebda
Correct logic for cuda; Not compiling
MatthijsBurgh Sep 30, 2025
5c01ab4
Ignore onnx models
MatthijsBurgh Sep 30, 2025
acd9b51
Make linking conditional based on CUDA
MatthijsBurgh Sep 30, 2025
38032f7
Fix conditional cuda
MatthijsBurgh Oct 14, 2025
8e69019
(actions) allow manual trigger
MatthijsBurgh Oct 14, 2025
06a18c8
Feature/refactoring yolo (#4)
MatthijsBurgh Oct 21, 2025
d29d018
included .vscode file configs
IasonTheodorou Oct 21, 2025
2977980
updated tests on cmake lists
IasonTheodorou Oct 21, 2025
daa5337
updated .vscode files to build with new CMake
IasonTheodorou Oct 21, 2025
f87e8ee
Working args
MatthijsBurgh Oct 21, 2025
7957dbf
Fix header extension
MatthijsBurgh Oct 23, 2025
70ca2ea
fix(CMake): enable catkin_lint again
MatthijsBurgh Oct 28, 2025
677bece
Fix header extension
MatthijsBurgh Oct 28, 2025
ad9cf19
Fix header locations
MatthijsBurgh Oct 28, 2025
90afade
Added optional argument for images (you need to provide it only for t…
IasonTheodorou Oct 28, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 13 additions & 0 deletions .github/workflows/main.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
name: CI

on: [push, pull_request, workflow_dispatch]

jobs:
tue-ci:
name: TUe CI - ${{ github.event_name }}
runs-on: ubuntu-latest
steps:
- name: TUe CI
uses: tue-robotics/tue-env/ci/main@master
with:
package: ${{ github.event.repository.name }}
17 changes: 17 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
build/
images/*
onnxruntime*/
onnxruntime/*
docker/*
CMakefile
CMakeCache.txt
CMakeFiles/*
cmake_install.cmake
Makefile
SPEED-SAM-C-TENSORRT/
sam_inference/model/FastSAM-x.onnx
mask*
segmentation_results*

# Models
*.onnx
19 changes: 19 additions & 0 deletions .vscode/c_cpp_properties_json
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
{
"configurations": [
{
"name": "Linux",
"includePath": [
"${workspaceFolder}/**",
"/usr/include/opencv4",
"${workspaceFolder}/sam_inference/inc",
"${workspaceFolder}/sam_inference/build"
],
"defines": [],
"compilerPath": "/usr/bin/gcc",
"cStandard": "c11",
"cppStandard": "c++17",
"intelliSenseMode": "linux-gcc-x64"
}
],
"version": 4
}
48 changes: 48 additions & 0 deletions .vscode/launch.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"version": "0.2.0",
"configurations": [
{
"name": "Debug Yolo Model",
"type": "cppdbg",
"request": "launch",
"program": "/home/amigo/ros/noetic/system/devel/lib/yolo_onnx_ros/test_yolo_onnx_ros", // Path to the executable
"args": [
"/home/amigo/ros/noetic/repos/github.com/tue-robotics/yolo_onnx_ros/data/yolo11m.onnx",
"/home/amigo/Documents/repos/hero_sam.bak/pipeline/build/images"
], // Add any command-line arguments for your program here
"stopAtEntry": false,
"cwd": "${workspaceFolder}/build", // Set the working directory
"environment": [],
"externalConsole": false,
"MIMode": "gdb",
"setupCommands": [
{
"description": "Enable pretty-printing for gdb",
"text": "-enable-pretty-printing",
"ignoreFailures": true
}
],
//"preLaunchTask": "build-yolo-project" // Ensure the project is built before debugging
},
{
"name": "Debug Yolo Test",
"type": "cppdbg",
"request": "launch",
"program": "/home/amigo/ros/noetic/system/devel/lib/yolo_onnx_ros/yolo_test", // Path to the executable
"args": [], // Add any command-line arguments for your program here
"stopAtEntry": false,
"cwd": "${workspaceFolder}/build", // Set the working directory
"environment": [],
"externalConsole": false,
"MIMode": "gdb",
"setupCommands": [
{
"description": "Enable pretty-printing for gdb",
"text": "-enable-pretty-printing",
"ignoreFailures": true
}
],
//"preLaunchTask": "build-yolo-project" // Ensure the project is built before debugging
}
]
}
77 changes: 77 additions & 0 deletions .vscode/settings.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,77 @@
{
"files.associations": {
"iostream": "cpp",
"cctype": "cpp",
"clocale": "cpp",
"cmath": "cpp",
"cstdarg": "cpp",
"cstddef": "cpp",
"cstdio": "cpp",
"cstdlib": "cpp",
"cstring": "cpp",
"ctime": "cpp",
"cwchar": "cpp",
"cwctype": "cpp",
"array": "cpp",
"atomic": "cpp",
"strstream": "cpp",
"bit": "cpp",
"*.tcc": "cpp",
"bitset": "cpp",
"chrono": "cpp",
"codecvt": "cpp",
"complex": "cpp",
"condition_variable": "cpp",
"cstdint": "cpp",
"deque": "cpp",
"list": "cpp",
"map": "cpp",
"set": "cpp",
"unordered_map": "cpp",
"unordered_set": "cpp",
"vector": "cpp",
"exception": "cpp",
"algorithm": "cpp",
"functional": "cpp",
"iterator": "cpp",
"memory": "cpp",
"memory_resource": "cpp",
"numeric": "cpp",
"optional": "cpp",
"random": "cpp",
"ratio": "cpp",
"regex": "cpp",
"string": "cpp",
"string_view": "cpp",
"system_error": "cpp",
"tuple": "cpp",
"type_traits": "cpp",
"utility": "cpp",
"fstream": "cpp",
"initializer_list": "cpp",
"iomanip": "cpp",
"iosfwd": "cpp",
"istream": "cpp",
"limits": "cpp",
"mutex": "cpp",
"new": "cpp",
"ostream": "cpp",
"shared_mutex": "cpp",
"sstream": "cpp",
"stdexcept": "cpp",
"streambuf": "cpp",
"thread": "cpp",
"cfenv": "cpp",
"cinttypes": "cpp",
"typeindex": "cpp",
"typeinfo": "cpp",
"variant": "cpp",
"compare": "cpp",
"concepts": "cpp",
"numbers": "cpp",
"semaphore": "cpp",
"stop_token": "cpp",
"*.txx": "cpp",
"filesystem": "cpp"
}
}
94 changes: 94 additions & 0 deletions .vscode/tasks.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,94 @@
{
"version": "2.0.0",
"tasks": [
{
"label": "build-sam-project",
"dependsOrder": "sequence",
"dependsOn": ["configure-sam", "build-sam-binary"],
"problemMatcher": ["$gcc"],
"group": "build"
},
{
"label": "build-pipeline-project",
"dependsOrder": "sequence",
"dependsOn": ["configure-pipeline", "build-pipeline-binary"],
"problemMatcher": ["$gcc"],
"group": "build"
},
{
"label": "build-yolo-project",
"dependsOrder": "sequence",
"dependsOn": ["configure-yolo", "build-yolo-binary"],
"problemMatcher": ["$gcc"],
"group": "build"
},
{
"label": "configure-sam",
"type": "shell",
"command": "cmake",
"args": [
"-DCMAKE_BUILD_TYPE=Debug",
"-S", "${workspaceFolder}/sam_inference",
"-B", "${workspaceFolder}/sam_inference/build"
],
"problemMatcher": ["$gcc"]
},
{
"label": "configure-pipeline",
"type": "shell",
"command": "cmake",
"args": [
"-DCMAKE_BUILD_TYPE=Debug",
"-S", "${workspaceFolder}/pipeline",
"-B", "${workspaceFolder}/pipeline/build"
],
"problemMatcher": ["$gcc"]
},
{
"label": "configure-yolo",
"type": "shell",
"command": "cmake",
"args": [
"-DCMAKE_BUILD_TYPE=Debug",
"-DCMAKE_POLICY_VERSION_MINIMUM=3.5",
"-S", "${workspaceFolder}/",
"-B", "${workspaceFolder}/build"
],
"problemMatcher": ["$gcc"]
},
{
"label": "build-sam-binary",
"type": "shell",
"command": "cmake",
"args": [
"--build",
"${workspaceFolder}/sam_inference/build",
"--config", "Debug"
],
"problemMatcher": ["$gcc"]
},
{
"label": "build-pipeline-binary",
"type": "shell",
"command": "cmake",
"args": [
"--build",
"${workspaceFolder}/pipeline/build",
"--config", "Debug"
],
"problemMatcher": ["$gcc"]
},
{
"label": "build-yolo-binary",
"type": "shell",
"command": "cmake",
"args": [
"--build",
"${workspaceFolder}/build",
"--config", "Debug"
],
"problemMatcher": ["$gcc"]
}

]
}
Loading
Loading