first commit

tests/models/image_classification_common.py

@@ -0,0 +1,9 @@
+def check_result_item_keys(result_item):
+    assert result_item.keys() == {
+        "input_path",
+        "page_index",
+        "input_img",
+        "class_ids",
+        "scores",
+        "label_names",
+    }