Commit f3a4a534 authored by Sam Tsai's avatar Sam Tsai Committed by Facebook GitHub Bot
Browse files

add assertion check in extended coco load

Summary: Dataset size check happened at https://fburl.com/code/ey18d081 . Adding this assertion to break and detect errors earlier.

Differential Revision: D33108979

fbshipit-source-id: c931a2167fce8ec0ce764365778d879280bd5af4
parent 914054ac
...@@ -254,6 +254,13 @@ def convert_to_dict_list(image_root, id_map, imgs, anns, dataset_name=None): ...@@ -254,6 +254,13 @@ def convert_to_dict_list(image_root, id_map, imgs, anns, dataset_name=None):
) )
) )
assert len(dataset_dicts) != 0, (
f"Loaded zero entries from {dataset_name} empty. \n"
f" Size of inputs (imgs={len(imgs)}, anns={len(anns)})\n"
f" Filtered of inputs (seg={num_instances_without_valid_segmentation},"
f" ={num_instances_without_valid_bounding_box}\n"
)
return dataset_dicts return dataset_dicts
......
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