公开(公告)号：CN117437410A

公开(公告)日：2024-01-23

申请号：CN202210801649.8

申请日：2022-07-08

Applicant: 哈尔滨工业大学(威海)

Inventor： 康兆翔 ,  张盛平

IPC: G06V10/26 ,  G06T7/194 ,  G06V10/82 ,  G06V10/42 ,  G06V10/44 ,  G06V10/52 ,  G06V10/80 ,  G06V20/70 ,  G06N3/0464 ,  G06N3/0455 ,  G06N3/084

Abstract: 本发明公开了一种应用于图像编辑的自动抠图方法，包括以下步骤：首先，收集Composition‑1k数据集和DUTS数据集；对数据集进行数据增强；生成U2‑Net生成三分图的语义分割网络；生成SwinTransformer的抠图网络；将U2‑Net和Swin Transformer级联生成端到端网络(自动抠图网络)；利用分块patch进行图像推理；最后图像编辑——得到前景，替换背景。