复现unet论文中的weight_loss,经过思考和资料搜索,成功解决了问题。
结果如下:
首先,将图片输入为灰度图(正常情况下是RGB三通道,不利于后续的处理)。在转变为灰度图后图片像素值就只有0(白)和255(黑)了。
这时候白色部分是细胞,黑色部分是背景。
使用measure.label函数实现连通域的计算(8连通,即像素点周围的标记值和像素点的标记值一致的则视为同一块)
计算完连通域后除重排序就可以算出来有多少个标记
计算不同连通域之间的距离,取距离最近和倒数第二近的带入公式计算,得到权值。
得到权值图后按权值图显示即可。
代码在https://download.csdn.net/download/dalao_whs/85045166