(label, class_id, class_num, iteration=[4, 10], cut_branch=True)
| 231 | |
| 232 | |
| 233 | def scribble4class(label, class_id, class_num, iteration=[4, 10], cut_branch=True): |
| 234 | label = (label == class_id) |
| 235 | sk_map = scrible_2d(label, iteration=iteration) |
| 236 | if cut_branch and class_id != 0: |
| 237 | cut = Cutting_branch() |
| 238 | for i in range(sk_map.shape[0]): |
| 239 | lab = sk_map[i] |
| 240 | if lab.sum() < 1: |
| 241 | continue |
| 242 | sk_map[i] = cut(lab, seg_lab=label[i]) |
| 243 | if class_id == 0: |
| 244 | class_id = class_num |
| 245 | return sk_map * class_id |
| 246 | |
| 247 | |
| 248 | def generate_scribble(label, iterations, cut_branch=True): |
no test coverage detected