Skip to content

Commit

Permalink
Merge pull request #482 from Purg/dev/yolo-v11-inference
Browse files Browse the repository at this point in the history
YOLO v11 inference
  • Loading branch information
Purg authored Jan 6, 2025
2 parents cc5bfe9 + 7b69812 commit 54c31ff
Show file tree
Hide file tree
Showing 7 changed files with 3,244 additions and 2,245 deletions.
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

0 comments on commit 54c31ff

Please sign in to comment.