fix(opendataset): correct wrong visual status for LIP dataset

PR Closed: #1026
Graviti-AI · Sep 27, 2021 · 5fe4663 · 5fe4663
1 parent c378364
commit 5fe4663
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorbay/opendataset/LIP/loader.py b/tensorbay/opendataset/LIP/loader.py
@@ -96,7 +96,7 @@ def _get_data(keypoints_info: List[str], image_path: str, parsing_path: str) ->
     keypoints = LabeledKeypoints2D()
     for x, y, v in chunked(islice(keypoints_info, 1, None), 3):
         keypoints.append(
-            Keypoint2D(float(x), float(y), int(v)) if x.isnumeric() else Keypoint2D(0, 0, 0)
+            Keypoint2D(float(x), float(y), 1 - int(v)) if x.isnumeric() else Keypoint2D(0, 0, 0)
         )
     label.keypoints2d = [keypoints]
     return data