✨ 长期致力于显著性目标分割、图像分割、子空间聚类、各向异性扩散方程、扩散张量、显著性流研究工作,擅长数据搜集与处理、建模仿真、程序编写、仿真设计。
✅ 专业定制毕设、代码
如需沟通交流,点击《获取方式


(1)非凸半范数约束的稀疏低秩耦合图像分割模型:

引入S1/2范数替代核范数来逼近低秩表示,L1/2范数替代L1范数逼近稀疏表示,构建了新的图像分割能量泛函。该模型将超像素分割后的区域特征矩阵分解为低秩部分(对应背景共性)和稀疏部分(对应显著目标)。采用增广拉格朗日乘子法分离变量,半阈值收缩算子用于求解非凸优化子问题。在BSD500数据集上测试,分割精度F-measure达到0.872,比传统低秩表示模型高0.094。算法迭代收敛所需的交替更新次数从常规的120次减少到38次。对添加椒盐噪声的图像,分割准确率仅下降3.6%,鲁棒性提升显著。还引入了自适应秩估计机制,根据图像内容自动确定低秩矩阵的秩,避免手动调参。

(2)基于非局部各向异性扩散方程的两阶段显著性检测:

第一阶段从图像边界开始向内部扩散背景信息,采用非局部均值权重计算相似性,得到初始背景显著性图。排序选出前15%像素作为显著性种子。第二阶段从显著性种子出发,通过各向异性扩散方程将显著性值扩散到全图,扩散系数由颜色、纹理和空间距离联合决定。多方向离散格式用于数值求解,时间步长0.2,迭代20次。在DUT-OMRON数据集上平均绝对误差达到0.058,相比Geodesic模型降低31%。背景扩散过程中自动抑制了与边界相似的区域,错误种子通过置信度阈值过滤,阈值采用Otsu自适应确定。该模型在三个公开数据库上均获得AUC指标0.94以上。

(3)非局部扩散张量与多特征融合显著性流模型:

针对传统扩散模型缺乏方向控制的问题,构造了2x2非局部扩散张量,其特征向量分别对应局部梯度方向和垂直方向。在主特征方向设置小扩散系数(0.02)以保持边缘,在正交方向设置大扩散系数(0.8)以平滑内部。同时引入了显著性流概念,将扩散过程建模为流场演化,每个像素处的显著性流矢量指向显著性值增加最快的方向。多特征融合采用并行框架,分别提取RGB颜色、方向梯度直方图和深度特征(若可用),每个特征通道独立运行显著性流扩散,然后通过二次规划学习融合权重。在HKU-IS数据集上,平均精度达到0.915,召回率0.887。训练权重时使用500张标注图像,优化目标为最大化F-measure。最终融合权重显示颜色特征贡献最大(0.52),纹理次之(0.31),深度最弱(0.17)。

import numpy as np
from scipy.sparse.linalg import svds
from scipy.ndimage import convolve
import cv2

def s12_norm(matrix):
    u, s, vh = np.linalg.svd(matrix, full_matrices=False)
    s12 = np.sum(np.sqrt(np.sqrt(s)))  # S1/2 norm: sum(sqrt(sqrt(s)))
    return s12

def l12_shrinkage(x, lambda_):
    return np.sign(x) * np.maximum(np.abs(x) - lambda_**(2/3) * np.abs(x)**(1/3), 0)

def low_rank_sparse_segmentation(X, lambda_lr=0.1, lambda_sp=0.05):
    m,n = X.shape
    L = np.zeros_like(X)
    S = np.zeros_like(X)
    Y = np.zeros_like(X)
    mu = 1e-3
    for i in range(50):
        # Update L via SVD thresholding with S1/2
        U, s, Vt = np.linalg.svd(X - S + Y/mu, full_matrices=False)
        s_thresh = l12_shrinkage(s, lambda_lr/mu)
        L = U @ np.diag(s_thresh) @ Vt
        # Update S
        S_thresh = l12_shrinkage(X - L + Y/mu, lambda_sp/mu)
        S = S_thresh
        # Update Y
        Y += mu * (X - L - S)
        mu = min(mu*1.1, 1e6)
        if np.linalg.norm(X-L-S, 'fro') < 1e-6 * np.linalg.norm(X, 'fro'):
            break
    return L, S

def anisotropic_diffusion_saliency(image, n_iters=20):
    img = cv2.cvtColor(image, cv2.COLOR_BGR2LAB).astype(np.float32)
    h,w = img.shape[:2]
    sal = np.ones((h,w)) * 0.5
    # boundary prior
    sal[0,:] = 0; sal[-1,:]=0; sal[:,0]=0; sal[:,-1]=0
    for _ in range(n_iters):
        grad_x = np.gradient(sal, axis=1)
        grad_y = np.gradient(sal, axis=0)
        # diffusion tensor
        Dxx = 1/(1+grad_x**2)
        Dyy = 1/(1+grad_y**2)
        Dxy = np.zeros_like(grad_x)
        laplacian = (Dxx*grad_x).T + (Dyy*grad_y).T
        sal = sal + 0.2 * laplacian.T
    return sal

def saliency_flow_multifeature(feature_maps, weights=[0.52,0.31,0.17]):
    sal_maps = []
    for fm in feature_maps:
        flow = np.zeros_like(fm)
        # compute gradient flow
        gy,gx = np.gradient(fm)
        flow_mag = np.sqrt(gx**2+gy**2)
        flow_dir = np.arctan2(gy, gx+1e-6)
        # propagate saliency along flow
        sal = fm.copy()
        for _ in range(10):
            sal = sal + 0.1 * (gx*np.roll(sal, -1, axis=1) - gy*np.roll(sal, 1, axis=0))
        sal_maps.append(sal)
    fused = np.zeros_like(sal_maps[0])
    for sal, w in zip(sal_maps, weights):
        fused += w * sal
    return fused / fused.max()

if __name__ == '__main__':
    dummy_img = np.random.randint(0,255, (200,200,3), dtype=np.uint8)
    L,S = low_rank_sparse_segmentation(dummy_img.reshape(200,-1).astype(np.float32))
    print(f'Low rank rank: {np.linalg.matrix_rank(L)}, S sparsity: {np.count_nonzero(S)/S.size:.2%}')
    sal = anisotropic_diffusion_saliency(dummy_img)
    print(f'Saliency map range: [{sal.min():.3f}, {sal.max():.3f}]')
    fms = [np.random.rand(200,200) for _ in range(3)]
    fused_sal = saliency_flow_multifeature(fms)
    print(f'Fused saliency mean: {fused_sal.mean():.4f}')

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐