"vscode:/vscode.git/clone" did not exist on "93b83677a7df38022483ed68c2a942fe964fd015"
Commit 80c18641 authored by Cheng-Yang Fu's avatar Cheng-Yang Fu Committed by Facebook GitHub Bot
Browse files

Add the fields which will be used in point-based modeling.

Summary:
Add the fields which will be used in point-based modeling.
- `point_coords` : indicates the point_coords in the image.
- `point_labels`: indicates the foreground or background points.

Differential Revision: D29532103

fbshipit-source-id: 9af6c9b049e1d05fd0d77909b09de1feec391ce9
parent e69e0ffe
...@@ -175,7 +175,16 @@ def convert_to_dict_list(image_root, id_map, imgs, anns, dataset_name=None): ...@@ -175,7 +175,16 @@ def convert_to_dict_list(image_root, id_map, imgs, anns, dataset_name=None):
obj = { obj = {
field: anno[field] field: anno[field]
# NOTE: maybe use MetadataCatalog for this # NOTE: maybe use MetadataCatalog for this
for field in ["iscrowd", "bbox", "bbox_mode", "keypoints", "category_id", "extras"] for field in [
"iscrowd",
"bbox",
"bbox_mode"
"keypoints",
"category_id",
"extras",
"point_coords",
"point_labels",
]
if field in anno if field in anno
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment