pyvino
pyvino copied to clipboard
filter and normalize pose points
- filter
body_parts_plus = iter(body_parts)
for x, y in zip(body_parts_plus, body_parts_plus):
points_bbox_df.loc[(points_bbox_df[x] < points_bbox_df['xmin']), x] = np.nan
points_bbox_df.loc[(points_bbox_df[x] > points_bbox_df['xmax']), x] = np.nan
points_bbox_df.loc[(points_bbox_df[y] < points_bbox_df['ymin']), y] = np.nan
points_bbox_df.loc[(points_bbox_df[y] > points_bbox_df['ymax']), y] = np.nan
- norm
points_bbox_norm_df = pd.DataFrame([])
body_parts_plus = iter(body_parts)
for x, y in zip(body_parts_plus, body_parts_plus):
values_x = (points_bbox_df[x] - points_bbox_df['xmin']) / points_bbox_df['xmax']
values_y = (points_bbox_df[y] - points_bbox_df['ymin']) / points_bbox_df['ymax']
points_bbox_norm_df[x] = values_x
points_bbox_norm_df[y] = values_y