pytorch 36 基于GT-seg标签生成伪seg标签模拟模型语义分割
创始人
2025-05-29 21:14:40

在论文CascadePSP中提到对seg标签进行扰动,生成二值的伪seg标签,在此对其代码进行分析和改动,实现生成多值的伪seg标签,可用于模拟全局语义分割模型的分割结果。

1、关键代码

1.1 modify_boundary

modify_boundary函数是整个代码的关键,该函数主要实现对二值图轮廓的扰动并生成扰动后的二值图。该代码可以保存到utils目录下,文件名为boundary_modification.py
函数实现流程如下:
1、获取所有轮廓
2、随机移除单个轮廓的部分轮廓像素点
3、对单个轮廓保留的轮廓像素点进行扰动(通过修改轮廓的坐标点实现)
4、根据坐标扰动后的轮廓重新绘制图像
5、对新的轮廓随机进行调整,并保证调整结果与原图重合度在一定iou内

import cv2
import numpy as np
import random
import mathtry:from utils.de_transform 

相关内容

热门资讯

4家“中国英伟达”抢着上市 订阅 快刀财经 ▲ 做您的私人商学院各显神通的中国AI芯片。作者:奇偶工作室来源: 奇偶工作室(ID...
“招商系”老将王颖获批担任招商... 近日,国家金融监督管理总局深圳监管局(下称“深圳金融监管局”)发布行政许可批复,正式核准王颖招商信诺...
平安人寿临时提案遭华夏幸福否决... 中国平安(601318.SH)与华夏幸福(600340.SH)之间的百亿纠葛再度升级。继12月17日...
MiniMax递表,把大模型公... 文 / 王浩纯来源 / 节点财经在香港中环的金融心脏地带,一场关于中国 AI 未来的资本竞速正在悄然...
年入10亿的网红按摩仪,要IP... “健康焦虑”这个赛道,挺魔幻的。作者 |渡尘来源 |投资家(ID:touzijias)“健康焦虑”这...