MCPcopy Create free account
hub / github.com/ChenhongyiYang/QueryDet-PyTorch / __init__

Method __init__

visdrone/mapper.py:36–48  ·  view source on GitHub ↗
(self, cfg, is_train=True)

Source from the content-addressed store, hash-verified

34 """
35
36 def __init__(self, cfg, is_train=True):
37
38 self.tfm_gens = build_transform_gen(cfg, is_train)
39 # fmt: off
40 self.img_format = cfg.INPUT.FORMAT
41 self.mask_on = False
42 self.mask_format = cfg.INPUT.MASK_FORMAT
43 self.keypoint_on = False
44 self.load_proposals = False
45 self.keypoint_hflip_indices = None
46 # fmt: on
47
48 self.is_train = is_train
49
50 def __call__(self, dataset_dict):
51

Callers

nothing calls this directly

Calls 1

build_transform_genFunction · 0.85

Tested by

no test coverage detected