diff --git a/deepdoc/vision/__init__.py b/deepdoc/vision/__init__.py index c2c964326..64afcaf72 100644 --- a/deepdoc/vision/__init__.py +++ b/deepdoc/vision/__init__.py @@ -13,6 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. # +import io import pdfplumber @@ -48,7 +49,10 @@ def init_in_out(args): pdf_pages(fnm) return try: - images.append(Image.open(fnm)) + fp = open(fnm, 'rb') + binary = fp.read() + fp.close() + images.append(Image.open(io.BytesIO(binary)).convert('RGB')) outputs.append(os.path.split(fnm)[-1]) except Exception: traceback.print_exc()