捕获到内存错误主动抛出中止程序,以释放显存
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user