捕获到内存错误主动抛出中止程序，以释放显存

docker-compose.yml

@@ -1,6 +1,6 @@
 x-env:
   &template
-  image: fcb_photo_review:1.14.7
+  image: fcb_photo_review:1.14.8
   restart: always
 
 x-review:

photo_mask/auto_photo_mask.py

@@ -160,6 +160,8 @@ def get_mask_layout(image, name, id_card_num):
                     result += _find_boxes_by_keys(ID_CARD_NUM_KEYS)
 
             return result
+    except MemoryError as e:
+        raise e
     except Exception as e:
         logging.error("涂抹时出错！", exc_info=e)
         return result

photo_review/auto_photo_review.py

@@ -48,6 +48,9 @@ def ie_temp_image(ie, ocr, image):
             ie_result = []
         else:
             ie_result = ie({"doc": temp_file.name, "layout": layout})[0]
+    except MemoryError as e:
+        # 显存不足时应该抛出错误，让程序重启，同时释放显存
+        raise e
     except Exception as e:
         logging.error("信息抽取时出错", exc_info=e)
     finally: