开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

这个不应该返回图片吗,为啥只有返回true 完全不会用啊,呜呜呜

这个不应该返回图片吗,为啥只有返回true
img_v2_dbe31930-9a79-40f3-853f-d9d9a390920g.jpg

展开
收起
vlrqcokjjjwv2 2023-07-19 11:19:21 146 0
3 条回答
写回答
取消 提交回答
  • 如果您期望从ONNX模型中获得图像输出而不是布尔值,以下是一种可能的解决方案:

    1. 在代码中找到output_dict字典,并确保其中包含正确的输出名称。这通常可以通过查看模型的输入和输出节点名称来确定。

    2. 如果模型输出的是图像数据(例如图像的像素值),您可以使用matplotlib或其他图像处理库来将数组转换回图像格式,并进行显示或保存。

    下面是示例代码的更新版本,展示如何将模型输出的图像转换为图片并显示出来:

    import numpy as np
    import json
    import onnxruntime
    import matplotlib.pyplot as plt
    
    # 加载ONNX模型和创建会话
    onnx_model_path = 'path/to/your/model.onnx'
    sess = onnxruntime.InferenceSession(onnx_model_path)
    
    # 准备输入数据
    input_data = np.array([1, 2, 3, 4], dtype=np.float32)
    input_name = sess.get_inputs()[0].name
    
    # 运行推理
    output = sess.run(None, {input_name: input_data})
    
    # 将输出转换为图像
    output_image = output[0]  # 假设输出是第一个元素
    output_image = np.squeeze(output_image)  # 如果有多个维度,请对其进行适当调整
    
    # 显示图像
    plt.imshow(output_image)
    plt.show()
    
    2023-07-24 14:09:16
    赞同 展开评论 打赏
  • 北京阿里云ACE会长

    似乎是在使用一个预训练的模型进行人像抠图,生成一张输出图片。代码中使用了modelscope.pipelines模块的pipeline方法来设置管道,Tasks.portrait matting参数指定了要执行的任务。管道的输出结果存储在result变量中,然后使用cv2.imwrite()函数将其保存为一张图片。然而,只返回True而非生成的图片,原因可能是模型或输入的图像存在问题,导致管道出错或生成了意外的结果。

    2023-07-19 16:21:59
    赞同 展开评论 打赏
  • 这个输出的图片在什么路径下能找到呢,或者在哪设置呢,一头雾水,不知道怎么用

    2023-07-19 12:23:02
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载