fixed pr comments

hrnn · hrnn · commit 2ec23f49a3dc · 2025-01-07T01:42:47.000-04:00
diff --git a/vision_agent/utils/video_tracking.py b/vision_agent/utils/video_tracking.py
@@ -275,7 +275,12 @@ def post_process(
             mask = rle_decode_array(detection["mask"])
             label = f"{detection['id']}: {detection['label']}"
             return_frame_data.append(
-                {"label": label, "mask": mask, "score": 1.0, "rle": detection["mask"]}
+                {
+                    "label": label,
+                    "mask": mask,
+                    "rle": detection["mask"],
+                    "score": 1.0,
+                }
             )
         return_data.append(return_frame_data)
 
@@ -291,9 +296,8 @@ def post_process(
             display_frame_data.append(
                 {
                     "label": obj["label"],
-                    "score": obj["score"],
                     "bbox": denormalize_bbox(obj["bbox"], image_size),
-                    "mask": obj["rle"],
+                    "rle": obj["rle"],
                 }
             )
             del obj["rle"]

Original file line number	Diff line number	Diff line change
`@@ -275,7 +275,12 @@ def post_process(`
`275`	`275`	`mask = rle_decode_array(detection["mask"])`
`276`	`276`	`label = f"{detection['id']}: {detection['label']}"`
`277`	`277`	`return_frame_data.append(`
`278`		`- {"label": label, "mask": mask, "score": 1.0, "rle": detection["mask"]}`
	`278`	`+ {`
	`279`	`+ "label": label,`
	`280`	`+ "mask": mask,`
	`281`	`+ "rle": detection["mask"],`
	`282`	`+ "score": 1.0,`
	`283`	`+ }`
`279`	`284`	`)`
`280`	`285`	`return_data.append(return_frame_data)`
`281`	`286`
`@@ -291,9 +296,8 @@ def post_process(`
`291`	`296`	`display_frame_data.append(`
`292`	`297`	`{`
`293`	`298`	`"label": obj["label"],`
`294`		`- "score": obj["score"],`
`295`	`299`	`"bbox": denormalize_bbox(obj["bbox"], image_size),`
`296`		`- "mask": obj["rle"],`
	`300`	`+ "rle": obj["rle"],`
`297`	`301`	`}`
`298`	`302`	`)`
`299`	`303`	`del obj["rle"]`