image_classification_common.py 209 B

123456789
  1. def check_result_item_keys(result_item):
  2. assert result_item.keys() == {
  3. "input_path",
  4. "page_index",
  5. "input_img",
  6. "class_ids",
  7. "scores",
  8. "label_names",
  9. }