问题描述
我正在尝试跨模块发送图像数据。
以下工作正常:
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
当作为我的 Detectron2 模型的输入时,此函数的结果 (pg_image
) 工作正常。
但是,当我发送 pg_image.tolist()
并在接收 (np.array(pg_image)
) 时转换回 np 数组并将其发送到我的 Detectron2 模型时,我不断收到以下错误:
RuntimeError: "upsample_bilinear2d_channels_last" not implemented for 'Long'
那么,如果将 np 数组转换为列表并转换回更改数据?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)