Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

YOLO v11 inference #482

Merged
merged 9 commits into from
Jan 6, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
18 changes: 18 additions & 0 deletions TRAIN_AND_RUN_README.md
Original file line number Diff line number Diff line change
Expand Up @@ -231,6 +231,8 @@ chmod a-w \
```

#### Generate Object Predictions in the Scene

##### YoloV7 Models
Note that the input COCO file is that which was generated in the previous step.
This is to ensure that all represented videos and image frames are predicted on
and present in both COCO files.
Expand All @@ -247,6 +249,22 @@ python-tpl/yolov7/yolov7/detect_ptg.py \
Additional debug outputs may optionally be generated.
See the `-h`/`--help` options for more details.

##### YOLO V11 Models
```
yolo_v11_inference_objects \
-i m5_xstat-activity_truth.coco.json \
-o m5_xstat-object_detections/object_detections-yolov11_baseline_xl_sgd_20241216.coco.json \
--model-hands ~/dev/darpa-ptg/angel_system/model_files/object_detector/hands_model.pt \
--hand-img-size 768 \
--model-objects /data/paul.tunison/data/darpa-ptg/yolo_v11_object_detection/m5_v0.56/runs/detect/baseline-xl/weights/best.pt \
--obj-img-size 640 \
-e hands \
--model-device cuda:0 \
--conf-thresh 0.307 \
--save-img m5_xstat-object_detections-v11_images \
--save-vid
```

#### Generate Pose Predictions
Note that the input COCO file is that which was generated in the
`Generate activity classification truth COCO file` section.
Expand Down
Loading