From fad4456ab51172cd0df2be4f60d1b802664e6984 Mon Sep 17 00:00:00 2001
From: openvino-book <180585215@qq.com>
Date: Tue, 11 Feb 2025 15:37:53 +0800
Subject: [PATCH] Update infer_cls.py

add exception handler to file open
---
 tools/infer_cls.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/tools/infer_cls.py b/tools/infer_cls.py
index 6c26ff4aa82..bf21bd533cf 100755
--- a/tools/infer_cls.py
+++ b/tools/infer_cls.py
@@ -65,10 +65,14 @@ def main():
     model.eval()
     for file in get_image_file_list(config["Global"]["infer_img"]):
         logger.info("infer_img: {}".format(file))
-        with open(file, "rb") as f:
-            img = f.read()
-            data = {"image": img}
-        batch = transform(data, ops)
+        try:
+            with open(file, "rb") as f:
+                img = f.read()
+                data = {"image": img}
+            batch = transform(data, ops)
+        except Exception as e:
+            logger.error(f"Error processing file {file}: {str(e)}")
+            continue
 
         images = np.expand_dims(batch[0], axis=0)
         images = paddle.to_tensor(images)