弱监督细分的边界框:全局约束接近完全监督
官方存储库 。 谈话记录可。
目录
很酷的把戏
要求(PyTorch)
要重现我们的实验,请执行以下操作:
python3.8 +
火炬1.5+
nibabel(仅在切片3D卷时)
西皮
NumPy
Matplotlib
Scikit图片
sh
用法
代码的最重要的部分被包含在bounds.py ,与定义BoxPriorBounds 。 ( BoxBounds用于限制框大小的BoxBounds度。)它会生成一个元组列表,每个元组都包含该段的掩码(一次热编码)及其下限( w或更低)。 然后,这些元组将由BoxPrior中的BoxPrior损失losses.py 。
简而言之,事情按以下顺序进行:
BoxPriorBounds生成一个(segment,width)对list[tuple[Tensor, Tensor] ;
数据加
2021-11-21 15:06:54
59KB
Python
1