如何推理单张图片

[leizhu1989]

大佬，请问如何一张一张推理不同尺寸大小的图片

[hhhFuture]

+1，我也想知道怎么推理单张图片

[Zzxiangyuqin]

解决了吗？我也遇到了这个问题

[hhhFuture]

解决了吗？我也遇到了这个问题

解决了。onnx的导出和onnx推理方式，在issue里面有大佬贴出来了，你搜索一下就行了。 推理单张图方式如下：

# -*- coding: utf-8 -*-
"""
@Time ： 2025/3/15 下午5:42
@Auth ： Future
@File ：image_matting.py
@IDE ：PyCharm
"""

from inference import convert_video
import torch
from model import MattingNetwork
from thop import profile

if __name__ == '__main__':

    # model = MattingNetwork('resnet50').eval()
    # model.load_state_dict(torch.load('rvm_resnet50.pth'))

    model = MattingNetwork('mobilenetv3').eval()
    model.load_state_dict(torch.load('rvm_mobilenetv3.pth'))

    convert_video(
        model,                           # 模型，可以加载到任何设备（cpu 或 cuda）
        input_source= r'E:\Project_RobustVideoMatting\images\src', # 视频文件，或图片序列文件夹。将要抠图的图放到该文件夹
        input_resize=(1280,720),       # [可选项] 缩放视频大小 (w,h)
        downsample_ratio=0.375,           # [可选项] 下采样比，若 None，自动下采样至 512px  越大就越可能把周围的弄西弄进来
        output_type='png_sequence',             # 可选 "video"（视频）或 "png_sequence"（PNG 序列）
        output_composition=r'E:\Project_RobustVideoMatting\images\dst',# 会导出4通道的透明背景的图片序列
        output_alpha="pha.mp4",          # [可选项] 会创建一个pha.mp4文件夹,文件夹存放的是掩码图
        output_foreground="fgr.mp4",     # [可选项] 会创建一个fgr.mp4文件夹,文件夹存放的是抠图后的前景
        output_video_mbps=4,             # 若导出视频，提供视频码率
        seq_chunk=12,                    # 设置多帧并行计算
        num_workers=1,                   # 只适用于图片序列输入，读取线程
        progress=True                    # 显示进度条
    )
    print("finished matting!")