From 3c790374c7683f84245a7a9024e17027c8d03d22 Mon Sep 17 00:00:00 2001 From: KBolashev Date: Sun, 25 Aug 2024 13:56:51 +0300 Subject: [PATCH] logic oopsie --- dagshub_annotation_converter/formats/yolo/pose.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dagshub_annotation_converter/formats/yolo/pose.py b/dagshub_annotation_converter/formats/yolo/pose.py index 3abe29d..b858a54 100644 --- a/dagshub_annotation_converter/formats/yolo/pose.py +++ b/dagshub_annotation_converter/formats/yolo/pose.py @@ -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()