(self, config, model, device, number_neighbors=20)
| 20 | |
| 21 | class EvaluationPitts(): |
| 22 | def __init__(self, config, model, device, number_neighbors=20) -> None: |
| 23 | self.config = config |
| 24 | self.number_neighbors = number_neighbors |
| 25 | self.model = model |
| 26 | self.device = device |
| 27 | if config.MODEL.TYPE == "Lidar": |
| 28 | self.trans_lidar = Augment_Point_Data(is_train=False) |
| 29 | elif config.MODEL.TYPE == "LiSPH": |
| 30 | self.trans_lidar_sph = Augment_SPH_Data(img_size=config.DATA.SPH_PROJ.IMAGE_SIZE, |
| 31 | is_train=False) |
| 32 | self.projection = LaserProjection(device=device, |
| 33 | top_size=config.DATA.BEV_PROJ.IMAGE_SIZE, |
| 34 | z_range=config.DATA.BEV_PROJ.Z_RANGE, |
| 35 | sph_size=config.DATA.SPH_PROJ.IMAGE_SIZE, |
| 36 | fov_range=config.DATA.SPH_PROJ.FOV) |
| 37 | else: |
| 38 | raise NotImplementedError('Please Try Predefined Types') |
| 39 | |
| 40 | def get_features_recall(self, traj_num, trans_IDX, rot_IDX, rand_idx=0): |
| 41 | database = [] |
nothing calls this directly
no test coverage detected