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

Squashed changes with YoloNASPose & Loss #1512

Merged
Merged
Changes from 1 commit
Commits
Show all changes
41 commits
Select commit Hold shift + click to select a range
238e41c
Introduce sample-centric keypoint transforms
BloodAxe Oct 2, 2023
efeb4ef
Cleanup leftovers
BloodAxe Oct 2, 2023
fa1f79d
Fixed way of checking transforms that require additional samples
BloodAxe Oct 3, 2023
88cef73
Docstrings
BloodAxe Oct 3, 2023
a29f58a
:attr -> :param
BloodAxe Oct 3, 2023
c797100
Added docs clarifying behavior of mosaic & mixup
BloodAxe Oct 3, 2023
7b1b9ee
Added docs clarifying behavior of mosaic & mixup
BloodAxe Oct 3, 2023
0d00668
Improved tests
BloodAxe Oct 3, 2023
a445917
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release
BloodAxe Oct 3, 2023
f388fcc
Additional docstrings & typing annotations
BloodAxe Oct 3, 2023
5b24d06
Merge remote-tracking branch 'origin/feature/SG-1060-yolo-nas-pose-re…
BloodAxe Oct 3, 2023
a7c97ce
Added missing additional_samples_count field
BloodAxe Oct 4, 2023
b84817d
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release
BloodAxe Oct 4, 2023
4a21438
KeypointsRemoveSmallObjects
BloodAxe Oct 4, 2023
53ecb3e
KeypointsRemoveSmallObjects
BloodAxe Oct 4, 2023
577804c
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release
BloodAxe Oct 5, 2023
fd41b71
Feature/sg 1060 yolo nas pose release pr to add datasets and metric (…
BloodAxe Oct 9, 2023
81091b7
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release
BloodAxe Oct 9, 2023
a1284be
Squashed changes with YoloNASPose & Loss
BloodAxe Oct 9, 2023
2d01377
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release-add-…
BloodAxe Oct 9, 2023
3ed26ff
Remove print statement
BloodAxe Oct 10, 2023
359dcf6
Fixed attribute name that was not renamed
BloodAxe Oct 10, 2023
928da84
Improve docstrings to use 'Num Keypoints' instead of magic number 17
BloodAxe Oct 10, 2023
0e540e5
Fixed PoseNMS export to work with custom number of keypoints
BloodAxe Oct 10, 2023
8dc7647
Added docstrings
BloodAxe Oct 10, 2023
0e23800
Simplify forward/forward_eval
BloodAxe Oct 10, 2023
15cef48
Simplify forward/forward_eval
BloodAxe Oct 10, 2023
295b0c7
_insert_heads_list_params
BloodAxe Oct 10, 2023
0474536
Merge master
shaydeci Oct 11, 2023
94a15fc
Added tests
BloodAxe Oct 11, 2023
74ea314
Refactor the way we generate usage instructions. Should be easier to …
BloodAxe Oct 11, 2023
d43cf2b
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release-add-…
BloodAxe Oct 12, 2023
408b5d0
Improve docstrings
BloodAxe Oct 12, 2023
e8b3f18
Improved docstrings
BloodAxe Oct 12, 2023
b2ff5e9
Improved docstrings
BloodAxe Oct 12, 2023
90bca43
Improved docstrings
BloodAxe Oct 12, 2023
84aa12b
Improved docstrings
BloodAxe Oct 12, 2023
fe00d56
Rename bboxes -> bboxes_xyxy
BloodAxe Oct 12, 2023
632d62d
Fixed instructions text
BloodAxe Oct 12, 2023
11d3eb2
Improve efficiency of training
BloodAxe Oct 12, 2023
91dd113
Merge branch 'master' into feature/SG-1060-yolo-nas-pose-release-add-…
BloodAxe Oct 13, 2023
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
Prev Previous commit
Next Next commit
Fixed instructions text
  • Loading branch information
BloodAxe committed Oct 12, 2023

Verified

This commit was signed with the committer’s verified signature.
BloodAxe Eugene Khvedchenya
commit 632d62d62055b213c95b58883bb83af35b8e8385
12 changes: 6 additions & 6 deletions src/super_gradients/module_interfaces/usage_instructions.py
Original file line number Diff line number Diff line change
@@ -98,9 +98,9 @@ def build_usage_instructions_for_pose_estimation(

if batch_size == 1:
usage_instructions += """
pred_bboxes = flat_predictions[1:5]
pred_scores = flat_predictions[5]
pred_joints = flat_predictions[6:].reshape(-1, 3)
pred_bboxes = flat_predictions[:, 1:5]
pred_scores = flat_predictions[:, 5]
pred_joints = flat_predictions[:, 6:].reshape((len(pred_bboxes), -1, 3))
for i in range(len(pred_bboxes)):
confidence = pred_scores[i]
x_min, y_min, x_max, y_max = pred_bboxes[i]
@@ -115,9 +115,9 @@ def build_usage_instructions_for_pose_estimation(
for current_sample in range({batch_size}):
predictions_for_current_sample = predictions[predictions[0] == current_sample]
print("Predictions for sample " + str(current_sample))
pred_bboxes = flat_predictions[1:5]
pred_scores = flat_predictions[5]
pred_joints = flat_predictions[6:].reshape(-1, 3)
pred_bboxes = predictions_for_current_sample[:, 1:5]
pred_scores = predictions_for_current_sample[:, 5]
pred_joints = predictions_for_current_sample[:, 6:].reshape((len(pred_bboxes), -1, 3))
for i in range(len(pred_bboxes)):
confidence = pred_scores[i]
x_min, y_min, x_max, y_max = pred_bboxes[i]