人像抠图学习笔记

目录

RobustVideoMatting 实时视频抠图

Modnet 预测脚本

人脸分割BiseNetV2

MODNet u2net:

MODNet 方法

---

RobustVideoMatting 实时视频抠图

Modnet 预测脚本

Modnet 效果有时比RobustVideoMatting好，在衣服分割时，

backbone 是mobile v2

gpu 512*512 速度22ms。

import os
import sys
import argparse
import time

import numpy as np
from PIL import Image

import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision.transforms as transforms

from src.models.modnet import MODNet


if __name__ == '__main__':
    # define cmd arguments
    parser = argparse.ArgumentParser()
    parser.add_argument('--input-path', type=str,default=r'C:\Users\Administrator\Pictures\mm\pics', help='path of