(mask)
| 100 | |
| 101 | |
| 102 | def norm_mask(mask): |
| 103 | c, h, w = mask.size() |
| 104 | for cnt in range(c): |
| 105 | mask_cnt = mask[cnt,:,:] |
| 106 | if(mask_cnt.max() > 0): |
| 107 | mask_cnt = (mask_cnt - mask_cnt.min()) |
| 108 | mask_cnt = mask_cnt/mask_cnt.max() |
| 109 | mask[cnt,:,:] = mask_cnt |
| 110 | return mask |
| 111 | |
| 112 | |
| 113 | def label_propagation(args, model, frame_tar, list_frame_feats, list_segs, mask_neighborhood=None): |
no test coverage detected