Method load_file_func

dataloader/pittsburgh.py:40–50

(self, filename)

        self.file_idxs = np.arange(0, len(self.queries.keys()))
    
    def load_file_func(self, filename):
        if self.config.MODEL.TYPE == "Lidar" or self.branch == "Lidar":
            pcd = np.asarray(o3d.io.read_point_cloud(filename + ".pcd").points)
            if pcd.shape[0] != 4096:
                raise ValueError('{}.pcd does not have sufficient points'.format(filename))
            pcd = pc_normalize(pcd)
            output = self.lidar_data_aug(pcd)
        elif self.config.MODEL.TYPE == "LiSPH" or self.branch == "LiSPH":
            sph_img = Image.open(filename + "_sph.png")
            output = self.lisph_data_aug(sph_img)
        return output
        
    def generate_pickles(self):
        if self.config.TRAINING.IS_TRAIN: