Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
62 commits
Select commit Hold shift + click to select a range
45007d9
[MTB-8 & MTB-12] Implement Bounding Box Ratio Cost and Euclidean-Boun…
crespofer Jun 2, 2025
0105b3b
[MTB-15][MTB-16] Implement Class Gate and Weighted Mean Cost Matrix F…
Adammouedden Jun 2, 2025
f2d03e3
[MTB-13] Implement Combined IoU-Euclidean-BBox Ratio Cost Function
david-orjuela Jun 2, 2025
a198b89
[MTB-8] Vectorize Operations and Return Matrix (D, T) instead of (T, D)
crespofer Jun 3, 2025
6cdf1a2
[MTB-7][MTB-8] Implement Euclidean Cost and IOU-Euclidean Cost Functions
justinUCF Jun 8, 2025
49ac667
[MTB-15] Updated Weighted Mean Cost & Class Gate Cost Matrices to be …
Adammouedden Jun 8, 2025
754a4cc
Restore model-team to pre-overwrite state
david-orjuela Jun 10, 2025
2908d1a
[MTB-11] Implemented combined IOU-Bounding Box Ratio Cost Matrix func…
david-orjuela Jun 8, 2025
16824f1
Resolve merge conflict: accidental overwrite recovery
david-orjuela Jun 8, 2025
73d785d
[MTB-14] Implement Element-Wise Average Cost
mbarbosa1 Jun 10, 2025
80e7ab6
[MTB-1] IoU Cost Matrix
GerardGz Jun 10, 2025
a436a05
[MTB-12] Implement Euclidean Distance Cost Matrix
david-orjuela Jun 10, 2025
edb4715
Merge branch 'model-team' of https://github.com/Autonomous-droneProje…
david-orjuela Jun 10, 2025
1e0d320
UPDATED V5
GerardGz Jun 10, 2025
2b7bf19
Updated V7 (#24)
GerardGz Jun 10, 2025
5cd16fb
Update 7.1 (#25)
GerardGz Jun 10, 2025
d538428
[Fix] Average Cost Remove Appearance Vectors, Other Small Edits
david-orjuela Jun 10, 2025
b4866e9
Merge branch 'model-team' of https://github.com/Autonomous-droneProje…
david-orjuela Jun 10, 2025
069f6af
[Fix] Remove Test Cases
david-orjuela Jun 10, 2025
2a51a85
[Fix] Euclidean distance used top-left corners; updated to use box ce…
david-orjuela Jun 11, 2025
0ac54e0
Delete kestrel directory
david-orjuela Jun 11, 2025
6c902d0
DO-Phase2 Implementation of Model
david-orjuela Jun 18, 2025
ac2f369
[MTB-19] Refactor all cost function signatures and internal logic to …
david-orjuela Jun 19, 2025
e046bbc
[MTB-22] Implement Forward Propogation Method
GerardGz Jun 20, 2025
e39ebd3
rename files
crespofer Jun 24, 2025
3ae2eb6
rename training file
crespofer Jun 24, 2025
5441154
create data loader file
crespofer Jun 24, 2025
5815233
rename dir variable
crespofer Jun 24, 2025
fc0cf17
[MTB-23][MTB-24] Scaffold initial Re-ID training pipeline
david-orjuela Jun 25, 2025
061d44d
Training function
Adammouedden Jul 3, 2025
53b6867
Turbo-train Market-1501 with AMP, fast I/O, early stop & TensorBoard
david-orjuela Jul 12, 2025
d07ef76
Updated Version of train.py
david-orjuela Jul 13, 2025
cae96d1
create yolo script to get bbox
crespofer Jul 14, 2025
9009257
updated training.py
Jul 14, 2025
7cba8fc
TrackingNode
Jul 24, 2025
387e299
Updated TrackingNode.py
Jul 24, 2025
82c02de
GitIgnore Updates
david-orjuela Jul 31, 2025
21292d9
Merge branch 'model-team' of https://github.com/Autonomous-droneProje…
david-orjuela Jul 31, 2025
9730746
Mahalanobis_gate
GerardGz Jul 31, 2025
d796c18
Co-authored-by: Gerard Gimenez <GerardGz@users.noreply.github.com>
Aug 5, 2025
889fe7f
Pushing master
Adammouedden Aug 6, 2025
b5edf35
`kestrel_msgs`: Removed vision_msgs, replaced with custom messages
david-orjuela Aug 6, 2025
63e8234
Updated det2bbox function
Aug 12, 2025
2d0069d
New Message Updates
david-orjuela Aug 12, 2025
1bab52a
Resolve merge: honor deepSORT/CNN folder rename, drop generated artif…
Adammouedden Aug 13, 2025
6dcd2d0
Creation of ROS Input Node
david-orjuela Aug 14, 2025
13f1df8
set up input node
crespofer Aug 14, 2025
d4f5177
Clustering node initial commit
Adammouedden Aug 14, 2025
f1039ce
Merge branch 'model-ROS' of https://github.com/Autonomous-droneProjec…
Adammouedden Aug 14, 2025
3ec344c
Completed clustering node
Adammouedden Aug 15, 2025
15d89b4
Fixed a bug in clustering node
Adammouedden Aug 15, 2025
5d4ef58
Clustering node improved, safeguard for an empty track array
Adammouedden Aug 15, 2025
e2c6b9e
Co-authored-by: Jalyn N George <JalynG12563@users.noreply.github.com>
Aug 15, 2025
25afa93
Merge branch 'model-ROS' of https://github.com/Autonomous-droneProjec…
Aug 15, 2025
03005b3
match tracks to raw img they were detected from
crespofer Aug 15, 2025
864896b
add track/image alignment, GUI display, and whitelist publishing
david-orjuela Aug 23, 2025
fa04902
Merge branch 'master' of https://github.com/Autonomous-droneProject/M…
david-orjuela Oct 1, 2025
e82e05b
Merge branch 'master' into model-ROS
Syndric Oct 1, 2025
6b4f5dd
fix: msg conflicts
Syndric Oct 1, 2025
c7d9e42
Updating kestrel_msgs, kestrel_clustering, kestrel_input, kestrel_tra…
david-orjuela Dec 22, 2025
4890bd4
Refactor tracking_node pipeline and cleanup tracking logic
david-orjuela Dec 31, 2025
03febd3
Refactored tracking, vectorizing cost matrix + integrating DataAssoci…
david-orjuela Dec 31, 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
7 changes: 0 additions & 7 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,10 +1,3 @@
# ignore all mp4/mov in the YOLO Demo subfolder
src/_archive_legacy/models/deepSORT/CNN/YOLO Demo/*.mp4
src/_archive_legacy/models/deepSORT/CNN/YOLO Demo/*.mov

# but keep this one file
!src/_archive_legacy/models/deepSORT/CNN/YOLO Demo/Kestrel_Tracking_demo_Clip.mp4

# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
Expand Down
15 changes: 15 additions & 0 deletions docs/Model Team/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
# Model Team Documentation
## Pipeline Overview
The model team is responsible for the tracking of people (PEV riders) infront of the drone.
- The pipeline starts by feeding the drone's camera information to the `kestrel_vision` ROS2 package.
- `kestrel_vision` contains `vision_node.py`, which creates the ROS2 node that subscribes to the camera, runs YOLO and our CNN, and publishes these detections (bounding box + feature embeddings).
- Next, `kestrel_tracking` contains `tracking_node.py`, which similarly creates the node that subscribes to the vision node's detections, and begins tracking them. This involves initiating a TrackManager and publishing the associated Tracks.

## Pipeline Specifics
### kestrel_vision

## ROS Topics/Services

## Key Parameters

## Debug Checklist
Binary file added models/best_model_checkpoint.pth
Binary file not shown.
Loading