MCPcopy Create free account
hub / github.com/OpenGVLab/HumanBench / masked_assignment

Function masked_assignment

PATH/core/models/decoders/losses/matcher.py:354–360  ·  view source on GitHub ↗
(c, mask=None)

Source from the content-addressed store, hash-verified

352
353
354def masked_assignment(c, mask=None) -> list:
355 if mask is None:
356 return linear_sum_assignment(c)
357 c[mask == 0] = 1e12
358 iSet, jSet = linear_sum_assignment(c)
359 keep = mask[iSet, jSet]
360 return iSet[keep], jSet[keep]
361
362
363def relation_net_assignment(c, alive_mask, cur_all_mask, valid_mask):

Callers 1

relation_net_assignmentFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected