1

我正在尝试跨模块发送图像数据。

以下工作正常：

def process_image(pdf_path, page_dimensions):
  pdf_path = get_pdf(pdf_path, False)
  pdf_name = os.path.basename(pdf_path)
  with tempfile.TemporaryDirectory() as path:
    conversion_time = time.time()
    chart_images = convert_from_path(
          pdf_path=pdf_path,
          dpi=300,
          fmt="jpg",
          output_file=os.path.basename(str(pdf_path)).split(".")[0],
          output_folder=path,
          use_pdftocairo=False,
          paths_only=True,
          thread_count=8,
      )
    pg_dim, pg_image_path = page_dimensions[0], chart_images[0]
    pg_image = cv2.imread(pg_image_path, cv2.IMREAD_UNCHANGED)
    pg_image = cv2.resize(pg_image, (pg_dim[1], pg_dim[2]))
  return pg_image

pg_image当作为我的 Detectron2 模型的输入时，此函数 ( ) 的结果可以正常工作。

但是，当我pg_image.tolist()在接收 () 时发送并转换回 np 数组np.array(pg_image)并将其发送到我的 Detectron2 模型时，我不断收到以下错误：

RuntimeError: "upsample_bilinear2d_channels_last" not implemented for 'Long'

那么，如果将 np 数组转换为列表并转换回更改数据？

4

0 回答 0