PaddleOCR/tests/predictors/image_classification_common.py

10 lines
209 B
Python
Raw Normal View History

def check_result_item_keys(result_item):
assert result_item.keys() == {
"input_path",
"page_index",
"input_img",
"class_ids",
"scores",
"label_names",
}