MCPcopy Create free account
hub / github.com/AIRMEC/HECTOR / collate_features

Function collate_features

extract_features.py:490–494  ·  view source on GitHub ↗
(batch)

Source from the content-addressed store, hash-verified

488 return img, coord, is_tile_kept
489
490def collate_features(batch):
491 # Item 2 is the boolean value from tile filtering.
492 img = torch.cat([item[0] for item in batch if item[2]], dim=0)
493 coords = np.vstack([item[1] for item in batch if item[2]])
494 return [img, coords]
495
496def mergedpatch_gen(features, coords, dist_threshold=4, corr_threshold = 0.6):
497

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected