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

This commit is contained in:
2024-11-28 10:27:44 +08:00
parent 9749577c5a
commit 8ea5420520
3 changed files with 6 additions and 1 deletions

View File

@@ -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