import torch import cv2 import numpy as np from PIL import Image from torchvision import transforms from model import MattingNetwork # RVM 的模型定义文件 # 加载模型 model = MattingNetwork("resnet50") model.load_state_dict(torch.load(r"D:\project\seg\jacke121-rvm_128_json\model_a\rvm_resnet50.pth", map_location="cpu")) model = model.eval() # 输入图像 # img = cv2.imread("input.jpg")[:, :, ::-1] # BGR -> RGB # img = cv2.imread(r"D:\project_2025\live2d\LayerDiffuse_DiffusersCLI-main\imgs\inputs\mm01.jpg") # BGR -> RGB img = cv2.imread(r"D:\soft\801.jpeg") # BGR -> RGB img = cv2.imread(r"D:\soft\805.png") # BGR -> RGB h, w, _ = img.shape # 转 tensor img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = img.astype(np.float32) / 255.0 # 归一化到 0-1 # 转 tensor: [H,W,3] -> [1,3,H,W] src = torch.from_numpy(img).permute(2, 0, 1).unsqueeze(0) src = src.cpu() # RVM 需要 recurrent states (init None) rec = [None] * 4 # 推理,RVM 输入为:src, downsample_ratio, *rec with torch.no_grad(): fgr, pha, *rec = model(src, downsample_ratio=0.25, *rec) # fgr: foreground,pha: alpha matte fgr = fgr[0].permute(1, 2, 0).cpu().numpy() pha = pha[0].permute(1, 2, 0).cpu().numpy() # 合成为 rgba rgba = np.concatenate([fgr, pha], axis=2) # [H,W,4] rgba = (rgba * 255).astype(np.uint8) Image.fromarray(rgba).save("output_rvm.png") print("Done: output_rvm.png")rvm抠图笔记
张小明
前端开发工程师
黑客入门必看|常见的5种网络攻防技术(非常详细)
现在的地球上,世界人口近80亿,每天发生的各种恶性攻击事件数以百万计。那网络世界更不必多说,网络攻防战几乎每时每刻都在发生。 如果说打架斗殴离咱们还远,那网络攻防战在你打开手机或电脑的时候就已经开始了。 为了让刚入门的…
大模型学习完全指南:从理论到实践的六大阶段资源包,助你成为AI开发专家
该文章提供了一份全面的大模型学习资源包,包含从基础理论到企业级应用的六大学习阶段:NLP基础、NLP实战、多模态大模型、RAG应用、Agent项目实战和面试辅导。资源包还包括人工智能论文合集、52个落地案例、100数据科学经典书籍和600行业研究报告…
【数据分析】分数阶混沌系统的数值解研究附Matlab代码
✅作者简介:热爱科研的Matlab仿真开发者,擅长数据处理、建模仿真、程序设计、完整代码获取、论文复现及科研仿真。 🍎 往期回顾关注个人主页:Matlab科研工作室 🍊个人信条:格物致知,完整Matlab代码获取及仿…
3分钟搞定!Windows 11 LTSC微软商店终极安装指南
3分钟搞定!Windows 11 LTSC微软商店终极安装指南 【免费下载链接】LTSC-Add-MicrosoftStore Add Windows Store to Windows 11 24H2 LTSC 项目地址: https://gitcode.com/gh_mirrors/ltscad/LTSC-Add-MicrosoftStore 还在为Windows 11 LTSC系统没有微软商店而…
31岁前国脚无球可踢 躺赚1.59亿身价暴跌
你知道英超豪门切尔西队内,薪水最高的球员是谁吗?是铁血队长里斯詹姆斯?还是队史标王恩佐费尔南德斯?答案或许让所有蓝军球迷心碎——是拉希姆斯特林。据英国媒体此前披露,为了在2022年夏天从曼城招揽这位英格兰前国脚…
基于Web的开放性实验项目管理系统的设计与实现任务书
陕西服装工程学院本科毕业设计(论文)任务书毕业论文(设计)题目: 基于Web的开放性实验项目管理系统的设计与实现 学 院: 信息工程学院 专…