优化异常处理

photo_mask.py

@@ -261,13 +261,12 @@ def get_mask_layout(image, contents):
         # layouts = get_ocr_layout(OCR, temp_file.name)
         try:
             layouts = DOC_PARSER.parse({"doc": temp_file.name})["layout"]
-        except MemoryError as me:
-            # 如果是显存溢出问题，则抛出
-            raise me
-        except Exception as e:
-            # 如果是没有识别到文字等问题，则继续
-            logging.warning("识别失败", exc_info=e)
+        except TypeError as te:
+            # 如果是类型错误，大概率是没识别到文字
             layouts = []
+        except Exception as e:
+            # 如果出现其他错误，抛出
+            raise e
         if not layouts:
             # 无识别结果
             return result