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

(fix) Shape and bbox error #19

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

GMBarra
Copy link

@GMBarra GMBarra commented Feb 3, 2023

Add a small code to fix an error relate to pose_cls_dataset_convert.py when some times this condition while len(pose_sequences[object_id]) - frame_start >= sequence_length_min: was not met and the code add an extra None array to data_array. This generate a problem in the E2E inference in Pose Classification.

if data_array is not None:
    data_arrays.append(data_array)

The other small code I added was in plot_e2e_inference.py. I just add this to draw the bbox without the presence of the variable "bbox" in the files:

#Fix bbox, using minmax joints point
x_list,y_list = zip(*joints)
x_min = min(x_list)
x_max = max(x_list)
y_min = min(y_list)
y_max = max(y_list)
bbox = [x_min,y_min,x_max,y_max]

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant