PaddleOCR/tests/predictors/image_classification_common.py

10 lines
209 B
Python

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