分类：mean

2020
10-08

可视化pytorch 模型中不同BN层的running mean曲线实例

加载模型字典逐一判断每一层，如果该层是bn的runningmean，就取出参数并取平均作为该层的代表对保存的每个BN层的数值进行曲线可视化fromfunctoolsimportpartialimportpickleimporttorchimportmatplotlib.pyplotaspltpth_path='checkpoint.pth'pickle.load=partial(pickle.load,encoding="latin1")pickle.Unpickler=partial(pickle.Unpickler,encoding="latin1")pretrained_dict=torch.load(pth_path,map_loc... 继续阅读 >

2020
10-07

python实现mean-shift聚类算法

本文实例为大家分享了python实现mean-shift聚类算法的具体代码，供大家参考，具体内容如下1、新建MeanShift.py文件importnumpyasnp#定义预先设定的阈值STOP_THRESHOLD=1e-4CLUSTER_THRESHOLD=1e-1#定义度量函数defdistance(a,b):returnnp.linalg.norm(np.array(a)-np.array(b))#定义高斯核函数defgaussian_kernel(distance,bandwidth):return(1/(bandwidth*np.sqrt(2*np.pi)))*np.exp(-0.5*((dist... 继续阅读 >