不寒767 2022-04-21 16:48 采纳率: 50%
浏览 104
已结题

python图像分割模型读取测试图片

img


这张图里的代码是构建了一个0向量,但是我想放测试图片进去,不想要这个0向量。怎么改?
参考这个预测文件

img


改成像这种形式

  • 写回答

5条回答 默认 最新

  • 小马各 2022-04-21 20:24
    关注

    torch.zeros()是创建0向量。
    参考一下这个吧。
    大概思路就是先给图片路径,然后加载图片,再可选地对图片进行处理,最后放进去模型就可以输出了。

    normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) # 数据归一化
    transform_test = transforms.Compose([
            transforms.Resize(size=(224, 224)),
            transforms.ToTensor(),
            normalize
            ])
    img_path = '图片路径'
    img = Image.open(img_path)
    img_tensor = transform_test(img).unsqueeze(0) #拓展维度
    input = img_tensor.cuda(non_blocking=True)
    output = model(input)
    

    如有帮助,希望采纳一下!!!

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(4条)

报告相同问题?

问题事件

  • 系统已结题 5月5日
  • 已采纳回答 4月27日
  • 赞助了问题酬金5元 4月21日
  • 创建了问题 4月21日

悬赏问题

  • ¥15 交换机和交换机之间的链路带宽以及主机带宽的理解
  • ¥15 ai创想家对战模式代码
  • ¥15 集合A由3个2行4列二维数组构成,从集合A中任意取一个二维数组元素、如果该二维数组元素的对应列位置的上、下两数都是奇数,而且仅有2个列是奇数/奇数,则该数组有意义,并放入集合B中打印输出。
  • ¥15 电信IPV6 无法外网访问吗
  • ¥15 有偿求效果比较好的遥感影像匹配的c++代码
  • ¥15 博主,你好,我下载了你的智能网联汽车辅助驾驶安全信息检测系统,现在不会运行,可以教我吗,
  • ¥15 怎么在excle输入下列公式
  • ¥15 Arduino,利用modbus的RS485协议,进行对外置的温湿度传感器进行数据读取
  • ¥15 vhdl+MODELSIM
  • ¥20 simulink中怎么使用solve函数?