diff --git a/docker-compose.yml b/docker-compose.yml index 77a672a..2cb30ce 100644 --- a/docker-compose.yml +++ b/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: diff --git a/photo_mask/auto_photo_mask.py b/photo_mask/auto_photo_mask.py index fd1c18d..89592e5 100644 --- a/photo_mask/auto_photo_mask.py +++ b/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 diff --git a/photo_review/auto_photo_review.py b/photo_review/auto_photo_review.py index 41520ac..808ad64 100644 --- a/photo_review/auto_photo_review.py +++ b/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: