update

2024-02-27 07:44:35 +00:00
parent 3527a4af47
commit 2d21d2d215
5 changed files with 5 additions and 6 deletions
--- a/src/models/ocr_model/utils/inference.py
+++ b/src/models/ocr_model/utils/inference.py
@@ -21,6 +21,7 @@ def inference(
    if isinstance(imgs_path[0], str):
        imgs = convert2rgb(imgs_path) 
    else:  # already numpy array(rgb format)
+        assert isinstance(imgs_path[0], np.ndarray)
        imgs = imgs_path 
    imgs = inference_transform(imgs)
    pixel_values = torch.stack(imgs)