我想从这个纸 (第19页)中计算python3 3/py手电筒中的表面骰子-S rensen系数。
我必须指出,我做的是而不是,尝试实现简单的标准体积骰子--S rensen系数!在我的实现中,这一条如下所示:
import torch
def volumetric_DSC(M1, M2):
M1 = M1.view(-1)
M2 = M2.view(-1)
dividend = 2 * (M1 * M2).sum()
divisor = (M1 * M1).sum() + (M2 * M2).sum()
return dividend / divisor
if __name__ == "__main__":
m1 = torch.empty(5, 5, 5).uniform_(0, 1)
m1 = torch.bernoulli(m1)
m2 = torch.empty(5, 5, 5).uniform_(0, 1)
m2 = torch.bernoulli(m2)
loss = volumetric_DSC(m1, m2)
print("loss = {0}".format(loss))如何将此代码扩展到表面骰子-S rensen系数损失?
https://stackoverflow.com/questions/56685144
复制相似问题