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

Update yolov7 submodule reference for script to output all classes #478

Merged
merged 2 commits into from
Nov 26, 2024
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
24 changes: 13 additions & 11 deletions TRAIN_AND_RUN_README.md
Original file line number Diff line number Diff line change
Expand Up @@ -137,13 +137,15 @@ bbn_create_truth_coco \
activity-truth-COCO.json
```


### Storage on Gyges
- On gyges, raw data is located at
- `/data/PTG/medical/bbn_data/Release_v0.5/v0.56/<task>`
- pre-trained models are available on `https://data.kitware.com/#collection/62cc5eb8bddec9d0c4fa9ee1/folder/6605bc558b763ca20ae99f55`
- In this pipeline we are only provided with object detection ground truth training data, which is located `/data/PTG/medical/object_anns/<task>`
- For real-time execution, we store our models in /angel_system/model_files
- Zed "Pro" data `/data/PTG/medical/bbn_data/Release_v0.5/v0.56/<task>`
- Hololens Golden data: `/data/PTG/medical/bbn_data/lab_data-golden/`
- Hololens Working data: `/data/PTG/medical/bbn_data/lab_data-working/`
- In this pipeline we are only provided with object detection ground truth
training data, which is located `/data/PTG/medical/object_anns/<task>`
- For real-time execution, we store our models in `angel_system/model_files/`
which are provisioned by Ansible by configuration.

##### Examples of files and what they are used for:

Expand Down Expand Up @@ -249,12 +251,12 @@ See the `-h`/`--help` options for more details.
Note that the input COCO file is that which was generated in the
`Generate activity classification truth COCO file` section.
```
python-tpl/TCN_HPL/tcn_hpl/data/utils/pose_generation/generate_pose_data.py \\
-i ~/data/darpa-ptg/bbn_data/lab_data-working/m2_tourniquet/activity_truth.coco.json \\
-o ./test_pose_output.coco.json \\
--det-config ./python-tpl/TCN_HPL/tcn_hpl/data/utils/pose_generation/configs/medic_pose.yaml \\
--det-weights ./model_files/pose_estimation/pose_det_model.pth \\
--pose-config ./python-tpl/TCN_HPL/tcn_hpl/data/utils/pose_generation/configs/ViTPose_base_medic_casualty_256x192.py \\
python-tpl/TCN_HPL/tcn_hpl/data/utils/pose_generation/generate_pose_data.py \
-i ~/data/darpa-ptg/bbn_data/lab_data-working/m2_tourniquet/activity_truth.coco.json \
-o ./test_pose_output.coco.json \
--det-config ./python-tpl/TCN_HPL/tcn_hpl/data/utils/pose_generation/configs/medic_pose.yaml \
--det-weights ./model_files/pose_estimation/pose_det_model.pth \
--pose-config ./python-tpl/TCN_HPL/tcn_hpl/data/utils/pose_generation/configs/ViTPose_base_medic_casualty_256x192.py \
--pose-weights ./model_files/pose_estimation/pose_model.pth
# Repeat for other relevant activity truth inputs
```
Expand Down
2 changes: 1 addition & 1 deletion python-tpl/yolov7
Submodule yolov7 updated 1 files
+0 −14 yolov7/detect_ptg.py
Loading