Skip to content

Commit

Permalink
logic oopsie
Browse files Browse the repository at this point in the history
  • Loading branch information
kbolashev committed Aug 25, 2024
1 parent 4a31648 commit 3c79037
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dagshub_annotation_converter/formats/yolo/pose.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ def export_pose(annotation: IRPoseImageAnnotation, context: YoloContext) -> str:
if context.keypoint_dim == 2:
point_list = [f"{point.x} {point.y}" for point in annotation.points if point.visible is not False]
else:
point_list = [f"{point.x} {point.y} {0 if point.visible is not False else 1}" for point in annotation.points]
point_list = [f"{point.x} {point.y} {0 if point.visible is False else 1}" for point in annotation.points]

category = annotation.ensure_has_one_category()

Expand Down

0 comments on commit 3c79037

Please sign in to comment.