Предварительно обученная модель Detectron2 с использованием LayoutParser в контейнере Docker. Ошибка: контрольная точка ⇐ Python

[Anonymous]

ниже приведен мой Dockerfile.

Код: Выделить всё

FROM python:3.9
RUN apt-get clean && apt-get update
pip install --upgrade pip

RUN pip install layoutparser

RUN pip install "layoutparser[ocr]"

RUN pip install pytesseract

RUN pip install pdf2image

RUN pip install torch

RUN pip install torchvision

RUN apt-get install -y poppler-utils  #(pdf-image)

RUN apt-get install -y tesseract-ocr

RUN apt-get install git #(to install detectron2)

RUN pip install "git+https://github.com/facebookresearch/detectron2.git"  #(detectron2 model)

RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y #(required to run packages)

workdir /home/jovyan/work/layout_parser

volume ["/home/jovyan/work/layout_parser"]

CMD ["python", "test_code.py"]

Код Python test_code.py:

Код: Выделить всё

import pdf2image
import layoutparser as lp
import pytesseract
import numpy as np
import cv2
import matplotlib.pyplot as plt

pdf_file= r"/home/jovyan/work/layout_parser/test_pdf.pdf"
image = np.asarray(pdf2image.convert_from_path(pdf_file)[0])

model = lp.Detectron2LayoutModel('lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config')

Я получил следующую ошибку:


Я безуспешно пытался решить проблему следующим образом:

• используемые варианты фонарика и версий torchvision< /li>
  Используется база Python 3.7, 3.8, 3.9
• Используются варианты config_path предварительно созданной модели
• Загружено вручную модель присутствует в файле конфигурации -> Layout_parser_modelzoo
• попробовал вручную загрузить модель с расширением pth и pkl. Получена следующая ошибка:
  

Как использовать предварительно обученные модели?

Подробнее здесь: https://stackoverflow.com/questions/760 ... rror-check